結果

問題 No.931 Multiplicative Convolution
ユーザー 👑 NachiaNachia
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

#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;
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0