【洛谷P4245】 【模板】任意模数NTT

三模数 NTT，感觉不是很难写 $?$         

代码借鉴的 https://www.cnblogs.com/Mychael/p/9297652.html

code: 

#include <bits/stdc++.h>            
#define SIZE 400005    
#define ll long long 
#define setIO(s) freopen(s".in","r",stdin)    
using namespace std;                 
inline ll qpow(ll x,ll y,ll Mod) 
{
    ll re=1ll;   
    for(x%=Mod;y;y>>=1,x=x*x%Mod)     if(y&1)   re=re*x%Mod;  
    return re;   
}     
int pr[3]={469762049,998244353,1004535809};          
struct poly 
{
    int G,Mod,A[SIZE];  
    void NTT(int *a,int len,int flag) 
    {
        int i,j,k,mid; 
        for(i=k=0;i<len;++i) 
        {
            if(i>k)    swap(a[i],a[k]);         
            for(j=len>>1;(k^=j)<j;j>>=1);   
        }         
        for(mid=1;mid<len;mid<<=1) 
        {
            int wn=qpow(G,(Mod-1)/(mid<<1),Mod);   
            if(flag==-1)   wn=qpow(wn,Mod-2,Mod);   
            for(i=0;i<len;i+=mid<<1) 
            {
                int w=1;  
                for(j=0;j<mid;++j) 
                {
                    int x=a[i+j], y=1ll*w*a[i+j+mid]%Mod;  
                    a[i+j]=1ll*(x+y)%Mod, a[i+j+mid]=(x-y+Mod)%Mod;  
                    w=1ll*w*wn%Mod;     
                }
            }
        }
        if(flag==-1)   
        {
            int rev=qpow(len,Mod-2,Mod);   
            for(i=0;i<len;++i)    a[i]=1ll*a[i]*rev%Mod;   
        }
    }       
}ntt[3];     
ll ans[SIZE];   
int F[SIZE],G[SIZE],B[SIZE],deg1,deg2,deg,md; 
ll INV(ll n,ll p) { return qpow(n%p, p-2,p); }     
ll mul(ll a,ll b,ll p) 
{   
    ll re=0;   
    for(;b;b>>=1,a=(a+a)%p)        if(b&1)   re=(re+a)%p;  
    return re;   
}
void CRT() 
{
    deg=deg1+deg2;   
    ll a,b,c,t,k,M=1ll*pr[0]*pr[1];   
    ll inv1=INV(pr[1],pr[0]),  inv0=INV(pr[0],pr[1]),inv3=INV(M%pr[2],pr[2]);        
    for(int i=0;i<=deg;++i)   
    {
        a=ntt[0].A[i];   
        b=ntt[1].A[i];   
        c=ntt[2].A[i];                                                                                                    
        t=(mul(a*pr[1]%M,inv1,M)+mul(b*pr[0]%M,inv0,M))%M;       
        k=((c-t%pr[2])%pr[2]+pr[2])%pr[2]*inv3%pr[2];          
        ans[i]=((k%md)*(M%md)%md+t%md)%md;                             
    }
}
void conv() 
{
    int n=1;   
    while(n<=(deg1+deg2)) n<<=1;   
    for(int u=0;u<=2;++u) 
    {
        ntt[u].G=3;  
        ntt[u].Mod=pr[u];  
        for(int i=0;i<=deg1;++i)   ntt[u].A[i]=F[i];  
        for(int i=0;i<=deg2;++i)   B[i]=G[i];  
        for(int i=deg2+1;i<n;++i)   B[i]=0;   
        ntt[u].NTT(ntt[u].A,n,1); 
        ntt[u].NTT(B,n,1);   
        for(int i=0;i<n;++i)    ntt[u].A[i]=1ll*ntt[u].A[i]*B[i]%pr[u];  
        ntt[u].NTT(ntt[u].A,n,-1);   
    }
}
int main() 
{ 
    // setIO("input");   
    scanf("%d%d%d",&deg1,&deg2,&md);   
    for(int i=0;i<=deg1;++i)   scanf("%d",&F[i]);   
    for(int i=0;i<=deg2;++i)   scanf("%d",&G[i]);   
    conv();  
    CRT();   
    for(int i=0;i<=deg;++i)    printf("%lld ",ans[i]); 
    return 0; 
}