【NTTテンプレート】[UOJ#34]多項式乗算

5189 ワード

原根と単位複根の差は多くなく,pの残り系の周りを一周したと考えられ,コードはFFTと類似している.uoj#34多項式乗算
#include
#include
#include
using namespace std;
#define MAXN 100000
#define MOD 998244353
#define G 3
int n,m,N,a[MAXN*3+10],b[MAXN*3+10];
void Read(int &x){
    char c;
    while(c=getchar(),c!=EOF)
        if(c>='0'&&c<='9'){
            x=c-'0';
            while(c=getchar(),c>='0'&&c<='9')
                x=x*10+c-'0';
            ungetc(c,stdin);
            return;
        }
}
void read(){
    Read(n),Read(m);
    int i;
    for(i=0;i<=n;i++)
        Read(a[i]);
    for(i=0;i<=m;i++)
        Read(b[i]);
    for(N=1;N<=m+n;N<<=1);
}
int quick_pow(int a,int b){
    int ret(1);
    while(b){
        if(b&1)
            ret=1ll*ret*a%MOD;
        a=1ll*a*a%MOD;
        b>>=1;
    }
    return ret;
}
void ntt(int *a,int N,int f){
    int i,j=0,t,k;
    for(i=1;i1;i++){
        for(t=N;j^=t>>=1,~j&t;);
        if(ifor(i=1;i1){
        t=i<<1;
        int wn=quick_pow(G,(MOD-1)/t);
        for(j=0;jint w=1;
            for(k=0;k1ll*w*wn%MOD){
                int x=a[j+k],y=1ll*w*a[j+k+i]%MOD;
                a[j+k]=(x+y)%MOD,a[j+k+i]=(x-y+MOD)%MOD;
            }
        }
    }
    if(f==-1){
        reverse(a+1,a+N);
        int inv=quick_pow(N,MOD-2);
        for(i=0;i1ll*a[i]*inv%MOD;
    }
}
void solve(){
    ntt(a,N,1);
    ntt(b,N,1);
    int i;
    for(i=0;i1ll*a[i]*b[i]%MOD;
    ntt(a,N,-1);
}
void print(){
    for(int i=0;iprintf("%d ",a[i]);
    printf("%d
"
,a[n+m]); } int main() { read(); solve(); print(); }

転載先:https://www.cnblogs.com/outerform/p/5921827.html