3527: [Zjoi2014]力
Time Limit: 30 Sec Memory Limit: 256 MBSec Special JudgeSubmit: 2417 Solved: 1435
[Submit][Status][Discuss]
Description
给出n个数qi,给出Fj的定义如下:
![](http://www.lydsy.com/JudgeOnline/upload/201701/11.jpg)
令Ei=Fi/qi,求Ei.
Input
第一行一个整数n。
接下来n行每行输入一个数,第i行表示qi。
n≤100000,0<qi<1000000000
Output
n行,第i行输出Ei。与标准答案误差不超过1e-2即可。
Sample Input
5
4006373.885184
15375036.435759
1717456.469144
8514941.004912
1410681.345880
4006373.885184
15375036.435759
1717456.469144
8514941.004912
1410681.345880
Sample Output
-16838672.693
3439.793
7509018.566
4595686.886
10903040.872
3439.793
7509018.566
4595686.886
10903040.872
HINT
Source
仔细观察这个式子 前半部分的Ei很符合卷积的性质,但是后半部分看的十分不友善
我们用f1[i]表示q[i],f2[i]表示q[n-i-1],a[i]表示(1.0/i/i);
那么E[i]前半部分=sigma qj/(j-i)^2可以看做f1[i]*a[j-i],一个赤裸裸的卷积
后半部分可以表示为f2[n-i-1]*a[j-i],只好像不像卷积啊,我们把f2[i]数组给反过来,这样和不就是一个定值了么
#include <bits/stdc++.h>
#define ll long long
using namespace std;
const double pi=acos(-1);
inline int read(){
int x=0;int f=1;char ch=getchar();
while(!isdigit(ch)) {if(ch=='-') f=-1;ch=getchar();}
while(isdigit(ch)) {x=x*10+ch-'0';ch=getchar();}
return x*f;
}
const int MAXN=1e6+10;
typedef complex <double> E;
E a[MAXN],f1[MAXN],f2[MAXN],w[MAXN];
int L,H,R[MAXN];
inline void FFT(E *a,int f){
for(int i=0;i<L;i++){
if(i<R[i]) swap(a[i],a[R[i]]);
}
for(int len=2;len<=L;len<<=1){
int l=len>>1;
E wn(cos(pi/l),f*sin(pi/l));
for(int i=1;i<l;i++) w[i]=w[i-1]*wn;
for(int st=0;st<L;st+=len){
for(int k=0;k<l;k++){
E x=a[st+k];E y=w[k]*a[st+l+k];
a[st+k]=x+y;a[st+k+l]=x-y;
}
}
}
if(f==-1){
for(int i=0;i<L;i++){
a[i]/=L;
}
}
}
int main(){
int n=read();w[0].real()=1;
for(int i=0;i<n;i++){
scanf("%lf",&f1[i].real());
f2[n-i-1]=f1[i];
}
for(int i=1;i<=n;i++){
a[i]=(1.0/i/i);
}
L=1;
while(L<=n+n) L<<=1,H++;
for(int i=0;i<L;i++){
R[i]=(R[i>>1]>>1)|((i&1)<<(H-1));
}
FFT(f1,1);FFT(a,1);FFT(f2,1);
for(int i=0;i<L;i++){
f1[i]=f1[i]*a[i];
f2[i]=f2[i]*a[i];
}
FFT(f1,-1);FFT(f2,-1);
for(int i=0;i<n;i++){
printf("%.3lf\n",f1[i].real()-f2[n-i-1].real());
}
return 0;
}