NTT——快速数论变换

其实就是FFT

我真的觉得应该先学NTT，感觉比FFT中各种怪力乱神的复数运算好理解多了。。。

原根

对于一个质数$p$，如果存在一个$q$使得$q^{i}modp(0\le i<p)$互不相同，那么称$q$为$p$的原根。

然后这个玩意就可以像复数单位根一样使了。

#include<bits/stdc++.h>
using namespace std;
const int N=2e6+5;
template<typename tn> void read(tn &a){
	tn x=0,f=1;char c=' ';
	for(;!isdigit(c);c=getchar()) if(c=='-') f=-1;
	for(;isdigit(c);c=getchar()) x=x*10+c-'0';
	a=x*f;
}
#define ilove(xxx) cout<<"I love "<<xxx<<endl
const int mod=998244353;
int a[N],b[N];
int n,m,r[N];
int limit=1,l=0;
int poww(int a,int b){
	int ans=1;
	while(b){
		if(b&1) ans=1ll*ans*a%mod;
		a=1ll*a*a%mod;
		b>>=1;
	}
	return ans;
}
void NTT(int *a,int type){
	for(int i=0;i<limit;i++)
		if(i<r[i]) swap(a[i],a[r[i]]);
	for(int mid=1;mid<limit;mid<<=1){
		int Wn=poww(3,(mod-1)/(mid<<1));
		if(type==-1) Wn=poww(Wn,mod-2);
		for(int R=mid<<1,j=0;j<limit;j+=R){
            int w=1;
            for(int k=0;k<mid;k++,w=1ll*w*Wn%mod){
                int x=a[j+k],y=1ll*w*a[j+mid+k]%mod;
                a[j+k]=1ll*(x+y)%mod;
                a[j+mid+k]=1ll*(x-y+mod)%mod;
            }
        }
	}
}
int main(){
	read(n);
	read(m);
	for(int i=0;i<=n;i++){
		read(a[i]);
		a[i]=1ll*(a[i]+mod)%mod;
	}
	for(int i=0;i<=m;i++){
		read(b[i]);
		b[i]=1ll*(b[i]+mod)%mod;
	}
	while(limit<=n+m) limit<<=1,l++;
	for(int i=0;i<limit;i++) r[i]=(r[i>>1]>>1)|((i&1)<<(l-1));
	NTT(a,1);
	NTT(b,1);
	for(int i=0;i<=limit;i++)
		a[i]=1ll*a[i]*b[i]%mod;
	NTT(a,-1);
	int inv=poww(limit,mod-2);
	for(int i=0;i<=n+m;i++) printf("%d ",1ll*a[i]*inv%mod);
	puts("");
	return 0;
}