#include<bits/stdc++.h>
using namespace std;
const int mod = (479<<21)+1;
const int g = 3; //原根
long long quick_mod(long long a,long long b)
{
long long ans=1;
while(b){
if(b&1)
ans=ans*a%mod;
a=a*a%mod;
b>>=1;
}
return ans;
}
void NTT(int n,long long a[],bool on=false) //长度为N (2的次数),默认正变换,逆变换加true
{
int r=0;
while((1<<++r)!=n);
for(int i=0; i<n; i++){
int tmp=0;
for(int j=0; j<r; j++)//蝴蝶操作
if(i&(1<<j))
tmp+=1<<(r-j-1);
if(i<tmp)
swap(a[i],a[tmp]);
}
for(int i=1; i<=r; i++){
int m=1<<i;
long long wn=quick_mod(g,(mod-1)/m);
for(int k=0; k<n; k+=m){
long long w=1;
for(int j=0; j<(m>>1); j++){
long long t,u;
t=w*(a[k+j+(m>>1)]%mod)%mod;
u=a[k+j]%mod;
a[k+j]=(u+t)%mod;
a[k+j+(m>>1)]=((u-t)%mod+mod)%mod;
w=w*wn%mod;
}
}
}
if(on){
for(int i=1; i<n>>1; i++)
swap(a[i],a[n-i]);
long long inv=quick_mod(n,mod-2);
for(int i=0; i<n; i++)
a[i]=a[i]%mod*inv%mod;
}
}
[模板] - 快速数论变换 - 航神的传家宝
最新推荐文章于 2022-05-02 22:10:54 发布