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;
}