FFT板子

FFT板子

---

• 代码

#include<bits/stdc++.h>
using namespace std;

typedef double db;
typedef complex<double> cplx;
const int N=(1<<22)+5;
const db PI=acos(-1.00);
int rev[N];

void dft(int n,cplx *t,int typ){
	for(int i=0;i<n;i++)
		if(i<rev[i])swap(t[i],t[rev[i]]);
	for(int step=1;step<n;step<<=1){
		cplx wn=cplx(cos(PI/step),typ*sin(PI/step));
		for(int i=step<<1,j=0;j<n;j+=i){
			cplx w=cplx(1.0,0);
			for(int k=0;k<step;k++,w*=wn){
				cplx x=t[j+k],y=t[j+k+step]*w;
				t[j+k]=x+y,t[j+k+step]=x-y;
			}
		}
	}
}
void fft(int P,cplx *A,cplx *B,cplx*C){
	dft(P,A,1);
	dft(P,B,1);
	for(int i=0;i<=P;i++)C[i]=A[i]*B[i];
	dft(P,C,-1);
}

cplx A[N],B[N],C[N];
int n,m;
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=0;i<=n;i++)scanf("%lf",&A[i].real());
	for(int i=0;i<=m;i++)scanf("%lf",&B[i].real());
	int p=1,l=0;
	while(p<=(n+m))p<<=1,l++;
	for(int i=0;i<p;++i)
		rev[i]=(rev[i>>1]>>1)|((i&1)<<(l-1));
	fft(p,A,B,C);
	for(int i=0;i<=m+n;i++){
		printf("%d ",int(C[i].real()/p+0.5));
	}
	return 0;
}