【模板】NTT

从\(dalao\)那里扒了个板过来

const int G=3, P=(119<<23)+1;//998244353
int n, m, l, r[Maxn], lim=1, f[Maxn], g[Maxn];
int pow(int x, int k){int res=1; while(k){if(k&1) res=1ll*res*x%P; x=1ll*x*x%P; k >>= 1;} return res;}
void NTT(int *A, int tp){
    for(int i=0; i < lim; i++) if(i < r[i]) swap(A[i], A[r[i]]);
    int gn, g, x, y;
    for(int mid=1, R=mid<<1 ; mid < lim; mid <<= 1){
        gn=pow(G, (P-1)/(mid<<1));
        for(int j=0; j < lim; j+=R){ g=1;
            for(int k=0; k < mid; k++, g=1ll*g*gn%P) 
                x=A[j+k], y=1ll*g*A[j+k+mid]%P, A[j+k]=(x+y)%P, A[j+k+mid]=(x-y+P)%P;
        }
    }
    if(tp == 1) return ;
    int inv=pow(lim, P-2); reverse(A+1, A+lim);
    for(int i=0; i < lim; i++) A[i]=1ll*A[i]*inv%P;
}
void solve(){
    n=read(); m=read();
    for(int i=0; i <= n; i++) f[i]=read(); for(int i=0; i <= m; i++) g[i]=read();
    while(lim <= n+m) lim<<=1, l++; for(int i=0; i < lim; i++) r[i]=r[i>>1]>>1|((i&1)<<l-1);
    NTT(f, 1); NTT(g, 1); for(int i=0; i < lim; i++) f[i]=1ll*f[i]*g[i]%P;
    NTT(f, -1); for(int i=0; i < n+m-1; i++) printf("%d ", f[i]);
}

转载于:https://www.cnblogs.com/zerolt/p/9308665.html