FFT模板

FFT模板

---

安利一下前辈的博客，写的真的好点击这里：从多项式乘法到快速傅里叶变换

#include<bits/stdc++.h>
using namespace std;
const int MAXL=22;
const int MAXN=1<<MAXL;
const double PI=acos(-1.0);
typedef complex<double> c_d;
c_d A[MAXN],B[MAXN],C[MAXN];
int rev[MAXN];//预处理翻转 
void transform(int n,c_d *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){
        c_d wn=c_d(cos(PI/step),typ*sin(PI/step));//主单位根 
        for(int i=step<<1,j=0;j<n;j+=i){
            c_d w=c_d(1.0,0);
            for(int k=0;k<step;k++,w*=wn){
                c_d 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,c_d *A,c_d *B,c_d *C){
    //DFT
    transform(p,A,1);
    transform(p,B,1);
    for(int i=0;i<=p;i++)C[i]=A[i]*B[i];
    //IDFT
    transform(p,C,-1);
}
int main(){
    int n,m;
    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;
}