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

倒过来就是卷积了 就可以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;
}