【BZOJ 3527】 [Zjoi2014]力|FFT

第二次写这个题 整理下就出卷积了

调试0.5h+ 太™爽了 这都可以

自己写的复数居然炸了

我肉眼看了3遍 没看出来！！！！！！！！！！！！！！！

犯过一次这错误了。。。。。。。。。。。。。。。。

 x.r*i+x.i*r 复制时注意下！！！！！！！！！！！！！！！！！！

#include <algorithm>
#include  <iostream>
#include    <string>
#include    <cstdio>
#include     <cmath>
using namespace std;
const int MAXN=1e5*6+1;
const double PI=acos(-1);
struct C {
	double r,i;
	C(double x=0,double y=0) {r=x;i=y;}
	C operator + (C x) {return C(r+x.r,i+x.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,r*x.i+i*x.r);}
};
int n,rev[MAXN];
C a[MAXN],b[MAXN];
void FFF(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*=2)
	{
		C wn(cos(2*PI/i),T*sin(2*PI/i));
		for(int j=0;j<n;j+=i)
		{
			C tmp,t,w(1,0);
			for(int k=0;k<i/2;k++)
			{
				tmp=x[k+j];
				t=w*x[k+j+i/2];
				x[k+j]=tmp+t;
				x[k+j+i/2]=tmp-t;
				w=w*wn;
			}
		}
	}
}
int main()
{
	cin>>n;int tmp=n;
	for(int i=0;i<n;i++) scanf("%lf",&a[i].r);
	for(int i=n-1-1;i>=0;i--) b[i].r=-1.0/(n-1-i)/(n-1-i),b[n-1+n-1-i]=-b[i].r;
	int m=n*3;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);
	FFF(a,1);FFF(b,1);
	for(int i=0;i<n;i++) a[i]=a[i]*b[i];
	FFF(a,-1);
	for(int i=0;i<tmp;i++) printf("%.3lf\n",a[tmp-1+i].r/n);
	return 0;
}