【模板】FFT&NTT

总算把坑给补上了
推荐一篇博文：https://blog.csdn.net/enjoy_pascal/article/details/81478582

FFT

#include<bits/stdc++.h>
using namespace std;
const int N=3e6+77;
const double pi=acos(-1);
int len=1,lg=0,r[N];
struct yjy
{
	double x,y;
	yjy(double a=0,double b=0):x(a),y(b){}
	yjy operator +(yjy a){return yjy(x+a.x,y+a.y);}
	yjy operator -(yjy a){return yjy(x-a.x,y-a.y);}
	yjy operator *(yjy a){return yjy(x*a.x-y*a.y,x*a.y+y*a.x);}
}w[N],b[N],a[N];
int n,m;
void dft(yjy *a,int n,int f)
{
	for(int i=0; i<n; i++) if(i<r[i]) swap(a[i],a[r[i]]);
	for(int i=2; i<=n; i<<=1)
	{
		int now=i>>1;
		yjy wn=w[i]; wn.y*=f;
		for(int j=0; j<n; j+=i)
		{
			yjy w=yjy(1,0),x,y;
			for(int k=j; k<j+now; k++,w=w*wn)
			{
				x=a[k],y=w*a[k+now];
				a[k]=x+y; a[k+now]=x-y;
			}
		}
	}
	if(f==-1) for(int i=0; i<n; i++) a[i].x/=n;
}
void FFT(yjy *a,yjy *b,int n,int m)
{
	dft(a,len,1); dft(b,len,1);
	for(int i=0; i<=len; i++) a[i]=a[i]*b[i];
	dft(a,len,-1);
	for(int i=0; i<=n+m; i++) 
		printf("%d ",int(a[i].x+0.5));
}
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=0,x; i<=n; i++) scanf("%d",&x),a[i].x=x;
	for(int i=0,x; i<=m; i++) scanf("%d",&x),b[i].x=x;
	while(len<=n+m)
	{
		lg++; len<<=1;
	}
	for(int i=0; i<len; i++) r[i]=(r[i>>1]>>1)|((i&1)<<(lg-1));
	for(int i=2; i<=len; i<<=1) w[i]=yjy(cos(2*pi/i),sin(2*pi/i));
	FFT(a,b,n,m);
}

NTT

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int N=3e6+77,g=3,mod=998244353;
const double pi=acos(-1);
int len=1,lg=0,r[N];
int n,m;
ll a[N],b[N];
ll power(ll x,ll t)
{
	ll b=1;
	while(t)
	{
		if(t&1) b=b*x%mod;
		x=x*x%mod; t>>=1;
	}
	return b;
}
void dft(ll *a,int n,int f)
{
	for(int i=0; i<n; i++) if(i<r[i]) swap(a[i],a[r[i]]);
	for(int i=2; i<=n; i<<=1)
	{
		int now=i>>1;
		ll wn=power(g,(mod-1)/i);
		if(f==-1) wn=power(wn,mod-2);
		for(int j=0; j<n; j+=i)
		{
			ll w=1,x,y;
			for(int k=j; k<j+now; k++,w=w*wn%mod)
			{
				x=a[k],y=w*a[k+now]%mod;
				a[k]=(x+y)%mod; a[k+now]=(x-y+mod)%mod;
			}
		}
	}
	if(f==-1) for(int i=0; i<n; i++) a[i]=a[i]*power(n,mod-2)%mod;
}
void NTT(ll *a,ll *b,int n,int m)
{
	dft(a,len,1); dft(b,len,1);
	for(int i=0; i<=len; i++) a[i]=a[i]*b[i]%mod;
	dft(a,len,-1);
	for(int i=0; i<=n+m; i++) 
		printf("%d ",a[i]);
}
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=0; i<=n; i++) scanf("%lld",&a[i]);
	for(int i=0; i<=m; i++) scanf("%lld",&b[i]);
	while(len<=n+m)
	{
		lg++; len<<=1;
	}
	for(int i=0; i<len; i++) r[i]=(r[i>>1]>>1)|((i&1)<<(lg-1));
//	for(int i=2; i<=len; i<<=1) w[i]=yjy(cos(2*pi/i),sin(2*pi/i));
	NTT(a,b,n,m);
}