結果
問題 | No.931 Multiplicative Convolution |
ユーザー |
👑 ![]() |
提出日時 | 2020-10-22 17:26:13 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
WA
|
実行時間 | - |
コード長 | 2,082 bytes |
コンパイル時間 | 1,891 ms |
コンパイル使用メモリ | 173,372 KB |
実行使用メモリ | 10,624 KB |
最終ジャッジ日時 | 2024-07-21 09:24:31 |
合計ジャッジ時間 | 5,068 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 2 WA * 12 |
ソースコード
#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 P;ULL PowPmodP[100000];ULL iPowPmodP[100000];vector<ULL> A, B;vector<ULL> C;int main() {cin >> P;for (ULL i = 0, p = 1; i < P; i++, p = p * 2 % 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;}