题意:
令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;
}