結果
問題 | No.931 Multiplicative Convolution |
ユーザー | 👑 Nachia |
提出日時 | 2020-10-22 17:31:21 |
言語 | C++14 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 205 ms / 2,000 ms |
コード長 | 2,331 bytes |
コンパイル時間 | 1,467 ms |
コンパイル使用メモリ | 172,876 KB |
実行使用メモリ | 10,624 KB |
最終ジャッジ日時 | 2024-07-21 09:24:36 |
合計ジャッジ時間 | 4,066 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
5,248 KB |
testcase_01 | AC | 2 ms
5,376 KB |
testcase_02 | AC | 2 ms
5,376 KB |
testcase_03 | AC | 1 ms
5,376 KB |
testcase_04 | AC | 2 ms
5,376 KB |
testcase_05 | AC | 2 ms
5,376 KB |
testcase_06 | AC | 3 ms
5,376 KB |
testcase_07 | AC | 19 ms
5,376 KB |
testcase_08 | AC | 186 ms
10,496 KB |
testcase_09 | AC | 141 ms
10,368 KB |
testcase_10 | AC | 191 ms
10,112 KB |
testcase_11 | AC | 143 ms
10,240 KB |
testcase_12 | AC | 102 ms
7,296 KB |
testcase_13 | AC | 205 ms
10,496 KB |
testcase_14 | AC | 192 ms
10,496 KB |
testcase_15 | AC | 191 ms
10,624 KB |
testcase_16 | AC | 186 ms
10,368 KB |
コンパイルメッセージ
main.cpp: In function 'ULL primroot(ULL)': main.cpp:61:1: warning: control reaches end of non-void function [-Wreturn-type] 61 | } | ^
ソースコード
#include<bits/stdc++.h> using namespace std; using LL = long long; using ULL = unsigned long long; #define rep(i,n) for(int i=0; i<(n); i++) ULL mpow(ULL a, ULL i, ULL M) { if (i == 0) return 1; ULL r = mpow(a * a % M, i >> 1, M); if (i & 1) r = r * a % M; return r; } void NTT(vector<ULL>& A, ULL M, ULL g) { int N = A.size(); for (int i = 0, j = 0; j < N; j++) { if (i < j) swap(A[i], A[j]); for (int k = N >> 1; k > (i ^= k); k >>= 1); } for (int i = 1; i < N; i <<= 1) { ULL q = mpow(g, (M - 1) / i / 2, M), qj = 1; rep(j, i) { for (int k = j; k < N; k += i * 2) { ULL l = A[k], r = A[k + i] * qj % M; A[k] = (l + r); if (A[k] >= M) A[k] -= M; A[k + i] = (l + M - r); if (A[k + i] >= M) A[k + i] -= M; } qj = qj * q % M; } } } struct Fuse { ULL M, G; }; vector<ULL> TNT(const vector<ULL>& A, const vector<ULL>& B, Fuse F) { int Z = 1; while (Z < A.size() + B.size()) Z <<= 1; vector<ULL> Ax(Z), Bx(Z); ULL iZ = mpow(Z, F.M - 2, F.M); rep(i, Z) Ax[i] = Bx[i] = 0; rep(i, A.size()) Ax[i] = A[i]; rep(i, B.size()) Bx[i] = B[i]; NTT(Ax, F.M, F.G); NTT(Bx, F.M, F.G); rep(i, Z) Ax[i] = Ax[i] * Bx[i] % F.M; NTT(Ax, F.M, mpow(F.G, F.M - 2, F.M)); rep(i, Z) Ax[i] = Ax[i] * iZ % F.M; return move(Ax); } const Fuse fuse = { 998244353,3 }; ULL primroot(ULL M) { for (int g = 2; g < M - 2; g++) { bool ok = true; ULL p = 1; rep(i, (M - 1) / 2) { p = p * g % M; if (p == 1) { ok = false; break; } } if (ok) return g; } } ULL P; ULL g; ULL PowPmodP[100000]; ULL iPowPmodP[100000]; vector<ULL> A, B; vector<ULL> C; int main() { cin >> P; g = primroot(P); for (ULL i = 0, p = 1; i < P; i++, p = p * g % P) PowPmodP[i] = p; rep(i, P - 1) iPowPmodP[PowPmodP[i]] = i; A.resize(P - 1); rep(i, P - 1) { ULL a; cin >> a; A[iPowPmodP[i + 1]] = a; } B.resize(P - 1); rep(i, P - 1) { ULL a; cin >> a; B[iPowPmodP[i + 1]] = a; } C = TNT(A, B, fuse); rep(i, P - 1) C[i] = (C[i] + C[i + P - 1]) % 998244353; rep(i, P - 1) { if (i) cout << " "; cout << C[iPowPmodP[i + 1]]; } cout << endl; return 0; }