代码

// luogu-judger-enable-o2
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cmath>
using namespace std;
const int N=3000005;
const int mod=998244353;
int r[N],l,n,m,A[N],B[N],w[N];
int ksm(int a,int k)
{
    int s=1,b=a;
    for(;k;k>>=1,b=1ll*b*b%mod)
        if(k&1) s=1ll*s*b%mod;
    return s;
}
void NTT(int *P,int op)
{
    for(int i=0;i<n;i++) if(i<r[i]) swap(P[i],P[r[i]]);
    for(int i=1;i<n;i<<=1)
    {
        int W=ksm(3,(mod-1)/(i<<1));//3是998244353的一个原根
        if(op<0) W=ksm(W,mod-2); w[0]=1;
        for(int j=1;j<i;j++) w[j]=1ll*w[j-1]*W%mod;
        for(int j=0,p=i<<1;j<n;j+=p)
            for(int k=0;k<i;k++)
            {
                int X=P[j+k],Y=1ll*w[k]*P[i+j+k]%mod;
                P[j+k]=(X+Y)%mod;P[i+j+k]=(X-Y+mod)%mod;
            }
    }
    if(op<0) for(int i=0,inv=ksm(n,mod-2);i<n;i++)
                 P[i]=1ll*P[i]*inv%mod;
}
int main()
{
    cin>>n>>m;
    for(int i=0;i<=n;i++) cin>>A[i];
    for(int i=0;i<=m;i++) cin>>B[i];
    m=n+m;for(n=1;n<=m;n<<=1) l++;l--;
    for(int i=0;i<n;i++) r[i]=(r[i>>1]>>1)|((i&1)<<l);
    NTT(A,1);NTT(B,1);for(int i=0;i<n;i++) A[i]=1ll*A[i]*B[i]%mod;NTT(A,-1);
    for(int i=0;i<=m;i++) printf("%d ",A[i]);
    return 0;
}

评测结果