[BZOJ2194] FFT卷积

fft板子易错点:
1、bit为最长的多项式的位数
2、maxn> (1<< bit)
3、fft中第一层循环i表示两个i长度的块并成一个 2i 长度的块,所以wn为 2i的单位根,j也是+=2i。
4、记得idft的情况
代码:

#include<iostream>
#include<cstdio>
#include<cmath>
#include<complex>
using namespace std;
const int maxn=280010;//maxn>2^(2*n) 
const double pi=acos(-1);
typedef complex<double> cd;
cd a[maxn],b[maxn];
int n,r[maxn];
void getrev(int bit)
{
    for(int i=0;i<(1<<bit);i++)
        r[i]=(r[i>>1]>>1)+((i&1)<<(bit-1));
}
void fft(cd *a,int n,int dft)
{
    for(int i=0;i<n;i++)
        if(i<r[i]) swap(a[i],a[r[i]]);
    for(int i=1;i<n;i<<=1)//two i-area combine to an 2*i-area
    {
        cd wn=exp(cd(0,dft*pi/i));//dft*2*pi/(2*i) 
        for(int j=0;j<n;j+=(i<<1))//j+=2*i
        {
            cd wnk=cd(1,0);
            for(int k=j;k<j+i;k++)
            {
                cd x=a[k],y=wnk*a[k+i];
                a[k]=x+y;
                a[k+i]=x-y;
                wnk*=wn;
            }
        }
    }
    if(dft==-1) for(int i=0;i<n;i++) a[i]/=n;
}
int main()
{
    scanf("%d",&n);
    int x,y;
    for(int i=0;i<n;i++)
        {scanf("%d%d",&x,&y);a[i]=x;b[n-1-i]=y;}
    int bit=0;
    while((1<<bit)<2*n-2)   bit++;//(1<<bit)>2^(2*n)
    getrev(bit);
    fft(a,(1<<bit),1);
    fft(b,(1<<bit),1);
    for(int i=0;i<(1<<bit);i++)
        a[i]*=b[i];
    fft(a,(1<<bit),-1);
    for(int i=n-1;i<=2*n-2;i++)
        printf("%d\n",(int)round(a[i].real()));
    return 0;
}
/*5
3 1
2 4
1 1
2 4
1 4*/
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值