P3811 【模板】乘法逆元

https://www.luogu.com.cn/record/91843036

Luogu/P3811/P3811.cpp

@@ -1,18 +1,29 @@
-#include <bits/stdc++.h>
+#include <iostream>
 
-using namespace std;
+using std::cin;
+using std::cout;
+const char endl = '\n';
 
-long long inv[3000005];
+const int N = 3e6 + 5;
+
+int n, p, inv[N];
 
 int main() {
+    std::ios::sync_with_stdio(false);
+    cin.tie(nullptr);
+
+    cin >> n >> p;
+
     inv[0] = 0;
     inv[1] = 1;
-    int n, p;
-    scanf("%d%d", &n, &p);
-    printf("1\n");
-    for (int i = 2; i <= n; ++i) {
-        inv[i] = (long long)p - (p / i) * inv[p % i] % p;
-        printf("%d\n", inv[i]);
+
+    cout << inv[1] << endl;
+
+    for (int i = 2; i <= n; i++) {
+        inv[i] = p - static_cast<long long>(p) / i * inv[p % i] % p;
+
+        cout << inv[i] << endl;
     }
+
     return 0;
 }