結果

問題 No.931 Multiplicative Convolution
ユーザー 👑 NachiaNachia
提出日時 2020-10-22 17:26:13
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 2,082 bytes
コンパイル時間 4,244 ms
コンパイル使用メモリ 172,224 KB
実行使用メモリ 10,496 KB
最終ジャッジ日時 2023-09-28 14:43:39
合計ジャッジ時間 7,841 ms
ジャッジサーバーID
（参考情報） 		judge14 / judge11
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
4,380 KB
testcase_01 AC 1 ms
4,380 KB
testcase_02 AC 1 ms
4,380 KB
testcase_03 AC 2 ms
4,376 KB
testcase_04 WA -
testcase_05 WA -
testcase_06 WA -
testcase_07 WA -
testcase_08 WA -
testcase_09 AC 156 ms
10,240 KB
testcase_10 WA -
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 WA -
testcase_16 WA -
権限があれば一括ダウンロードができます

ソースコード

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;
}
0