您的位置:首页 > 其它

【BZOJ 2194】 快速傅立叶之二|FFT

2016-03-23 15:37 525 查看
倒过来就是卷积了 就可以FFT了

弱智错误 

1. /2

2. +0.5

3. <n

4. MAXN*4

#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdio>
#include <cmath>
using namespace std;
const double PI=acos(-1);
const int MAXN=400000+10;
struct C
{
double r,i;
C(double a=0,double b=0){r=a;i=b;}
C operator + (C x) {return C(x.r+r,x.i+i);}
C operator - (C x) {return C(r-x.r,i-x.i);}
C operator * (C x) {return C(r*x.r-i*x.i,x.i*r+x.r*i);}
};
int n,rev[MAXN];C a[MAXN],b[MAXN];
void FFT(C x[],int T)
{
for(int i=0;i<n;i++) if(rev[i]<i) swap(x[i],x[rev[i]]);
for(int i=2;i<=n;i<<=1)
{
C wn(cos(2*PI/i),T*sin(2*PI/i));
for(int j=0;j<n;j+=i)
{
C t,tmp,w(1,0);
for(int k=0;k<i/2;k++)
{
tmp=x[j+k];
t=w*x[j+k+i/2];
x[j+k]=tmp+t;
x[j+k+i/2]=tmp-t;
w=w*wn;
}
}
}
}
int main()
{
cin >>n; int tmp=n;
for(int i=0;i<n;i++) scanf("%lf %lf",&a[i].r,&b[n-i-1].r);
int m=n+n+1;n=1;
while(n<=m) n<<=1;
for(int i=0;i<n;i++) rev[i]= (rev[i>>1]>>1)|((i&1)?n>>1:0);
FFT(a,1);FFT(b,1);
for(int i=0;i<n;i++) a[i]=a[i]*b[i];
FFT(a,-1);
for(int i=0;i<tmp;i++) printf("%d\n",(int)(a[i+tmp-1].r/n+0.5));
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: