結果

問題 No.931 Multiplicative Convolution
コンテスト
ユーザー Nachia
提出日時 2020-10-22 17:31:21
言語 C++14
(gcc 13.3.0 + boost 1.87.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
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 14
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function 'ULL primroot(ULL)':
main.cpp:61:1: warning: control reaches end of non-void function [-Wreturn-type]
   61 | }
      | ^

ソースコード

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