快速数论变换 NTT

快速数论变换 NTT


和FFT很像。

typedef long long ll;
const int P=998244353;

inline ll ksm(ll a,ll n){
	ll ans=1;
	for(;n;n>>=1,a=a*a%P) if(n&1) ans=ans*a%P;
	return ans;
}

void NTT(int n,ll a[],int f){
    for(int i=0;i<n;i++) if(i<r[i]) swap(a[i],a[r[i]]);
    for(int i=1;i<n;i<<=1){
        ll wn=ksm(3,f==1?(P-1)/(i<<1):(P-1-(P-1)/(i<<1)));
        for(int j=0;j<n;j+=i<<1){
            ll w=1;
            for(int k=0;k<i;k++,w=w*wn%P){
                ll x=a[j+k],y=w*a[j+k+i]%P;
                a[j+k]=(x+y)%P,a[j+k+i]=(x-y+P)%P;
            }
        }
    }
    if(f==-1) for(int i=0;i<n;i++) a[i]=(a[i]*_n)%P;
}


int init(int n,int m){
    int l=0;
    m+=n;
    for(n=1;n<=m;n<<=1) l++;
    for(int i=0;i<n;i++) r[i]=(r[i>>1]>>1)|((i&1)<<(l-1));
    _n=ksm(n,P-2);
    return n;
}
 

##练习## - 【BZOJ3992】【SDOI2015】序列统计