BZOJ 3527: [Zjoi2014]力 [FFT]

题意:

令Ei=Fi/qi,求Ei.

思路:

首先注意E的下标是i,F的下标是i。然后化简一下这个式子。

#include <iostream>
#include <string.h>
#include <algorithm>
#include <stdio.h>
#include <math.h>
using namespace std;
#define N 262200+10
typedef long long ll;
const double PI =acos(-1.0);

struct Complex{
	double x,y;
	Complex(double _x=0.0,double _y=0.0)
	{
		x=_x;y=_y;
	}
	Complex operator -(const Complex &b)
	{
		return Complex(x-b.x,y-b.y);
	}
	Complex operator +(const Complex &b)
	{
		return Complex(x+b.x,y+b.y);
	}
	Complex operator *(const Complex &b)
	{
		return Complex(x*b.x-y*b.y,x*b.y+y*b.x);
	}
};
void change(Complex y[],int len)
{
	for(int i=1,j=len/2;i<len-1;i++)
	{
		if(i<j) swap(y[i],y[j]);
		int k=len/2;
		while(j>=k)
		{
			j-=k;k/=2;
		}
		if(j<k)
		{
			j+=k;
		}
	}
}
void fft(Complex y[],int len,int on)
{
	change(y,len);
	for(int h=2;h<=len;h<<=1)
	{
		Complex wn(cos(-on*2*PI/h),sin(-on*2*PI/h));
		for(int j=0;j<len;j+=h)
		{
			Complex w(1,0);
			for(int k=j;k<j+h/2;k++)
			{
				Complex u=y[k];
				Complex t=w*y[k+h/2];
				y[k]=u+t;
				y[k+h/2]=u-t;
				w=w*wn;
			}
		}
	}
	if(on==-1)
	{
		for(int i=0;i<len;i++)
		y[i].x/=len;
	}
}
void init(int &len,int &len1,int &len2)
{
	len=1;
	while(len<=len1*2||len<=len2*2)
	len<<=1;
}
Complex x1[N],x2[N],x3[N];
double a[N];
double ans1[N],ans2[N];
int main()
{
	int n;
	scanf("%d",&n);
	for(int i=0;i<n;i++) scanf("%lf",&a[i]);
	 
	for(int i=0;i<n;i++)
	x1[i].x=a[i],x1[i].y=0;
	
	for(int i=1;i<n;i++)
	x2[i].x=1.0/i/i,x2[i].y=0;
	
	int len,len1=n,len2=n;
	init(len,len1,len2);
	
	fft(x1,len,1);
	fft(x2,len,1);
	
	for(int i=0;i<len;i++)
	x1[i]=x1[i]*x2[i];
	fft(x1,len,-1);
	
	for(int i=0;i<n;i++) ans1[i]=x1[i].x;			//卷积1 
	
	for(int i=0;i<len;i++) x1[i]=Complex(),x2[i]=Complex();
	 
	for(int i=0;i<n;i++)
	x1[i].x=a[n-1-i],x1[i].y=0;
	for(int i=1;i<n;i++)
	x2[i].x=1.0/i/i,x2[i].y=0;
	
	fft(x1,len,1);
	fft(x2,len,1);
	
	for(int i=0;i<len;i++)
	x1[i]=x1[i]*x2[i];
	fft(x1,len,-1);
	
	for(int i=0;i<n;i++) ans2[i]=x1[n-1-i].x;			//卷积2 
	
	for(int i=0;i<n;i++) printf("%.3lf\n",ans1[i]-ans2[i]);
	
	return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_37632935/article/details/81482521