FFT模板

const double pi=acos(-1);
struct cplx{
    double r,i;//r为系数的实部,i为虚部。
    cplx(double r=0,double i=0):r(r),i(i){}
    cplx operator +(const cplx &a)const
    {return cplx(r+a.r,i+a.i);}
    cplx operator -(const cplx &a)const{
    return cplx(r-a.r,i-a.i);}
    cplx operator *(const cplx &a)const
    {return cplx(r*a.r-i*a.i,r*a.i+i*a.r);}    
}f1[maxn*4],f2[maxn*2],res[maxn*4];
void rader(cplx *f,int len){
    int j=len>>1;
    for(int i=1;i<len-1;++i){
    if(i<j)swap(f[i],f[j]);
    int k=len>>1;
    while(j>=k){
        j-=k;
        k>>=1;    
    }   
    if(j<k)j+=k;
    }    
}
void FFT(cplx *f,int len,int t){  
    rader(f,len);  
    for(int h=2;h<=len;h<<=1){  
    cplx wn(cos(-t*2*pi/h),sin(-t*2*pi/h));  
    for(int j=0;j<len;j+=h){  
        cplx e(1,0);  
        for(int k=j;k<j+h/2;++k){  
        cplx u=f[k];  
        cplx v=e*f[k+h/2];       
        f[k]=u+v;  
        f[k+h/2]=u-v;  
        e=e*wn;  
        }  
    }     
    }      
    if(t==-1)for(int i=0;i<len;i++)f[i].r/=len;  
}  
void Conv(cplx *a,cplx *b,int len){  
    FFT(a,len,1);  
    FFT(b,len,1);  
    for(int i=0;i<len;++i)a[i]=a[i]*b[i];  
    FFT(a,len,-1);      
}//a为第一个二项式系数,b为第二个二项式系数,len为长度。len的计算方法如下:设原本的二项式长度为n,len=pow(2,k)>n的第一个k的值

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值