3527: [Zjoi2014]力
Time Limit: 30 Sec Memory Limit: 256 MBSec Special Judge
Submit: 863 Solved: 532
[Submit][Status][Discuss]
Description
给出n个数qi,给出Fj的定义如下:
令Ei=Fi/qi,求Ei.
Input
第一行一个整数n。
接下来n行每行输入一个数,第i行表示qi。
Output
n行,第i行输出Ei。
与标准答案误差不超过1e-2即可。
Sample Input
5
4006373.885184
15375036.435759
1717456.469144
8514941.004912
1410681.345880
Sample Output
-16838672.693
3439.793
7509018.566
4595686.886
10903040.872
Hint
对于30%的数据,n≤1000。
对于50%的数据,n≤60000。
对于100%的数据,n≤100000,0
Ei=∑i>jqj(i−j)2−∑i<jqj(j−i)2
Ei=∑j=1i−1qj(i−j)2−∑j=i+1nqj(j−i)2
Ei=∑j=1i−1a[j]∗b[i−j]−∑j=i+1na[j]∗b[j−i]
a[i]=qi,b[i]=1i2
第一个式子已经化成了卷积的形式,对于第二个式子可以换一下元
∑j=0n−1−ia[n−1−j]∗b[j−i]
这样就也成了卷积的形式,然后做两遍FFT就行了。
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
using namespace std;
const int M=1000010;
int N,n,L,dig[M],rev[M];
double qi[M],ans[M];
struct S{
double x,y;
void prepare(double xx,double yy){x=xx;y=yy;}
S operator + (const S&xx){return (S){x+xx.x,y+xx.y};}
S operator - (const S&xx){return (S){x-xx.x,y-xx.y};}
S operator * (const S&xx){return (S){x*xx.x-y*xx.y,x*xx.y+y*xx.x};}
}a[M],b[M],c[M],d[M];
inline void FFT(S *a,int f){
int i,j,k;
S wn,x,y,w;
for(i=0;i<N;++i) d[i]=a[rev[i]];
for(i=0;i<N;++i) a[i]=d[i];
for(i=2;i<=N;i<<=1){
wn.prepare(cos(2*M_PI/i),f*sin(2*M_PI/i));
for(j=0;j<N;j+=i){
w.prepare(1,0);
for(k=j;k<j+i/2;++k){
x=a[k];
y=a[k+i/2]*w;
a[k]=x+y;
a[k+i/2]=x-y;
w=w*wn;
}
}
}
if(f==-1) for(i=0;i<N;++i) a[i].x/=(double)N*1.0;
}
int main(){
int i,j,len;
scanf("%d",&n);L=0;
for(N=1;N<n;N<<=1,L+=1); N<<=1;L+=1;
for(i=0;i<N;++i){
for(j=i,len=0;j;j>>=1) dig[len++]=j&1;
for(j=0;j<L;++j) rev[i]=rev[i]*2+dig[j];
}
for(i=0;i<n;++i) scanf("%lf",&qi[i]);
for(i=0;i<n;++i) a[i].prepare(qi[i],0);
for(i=1;i<n;++i) b[i].prepare(1.0/(double)i/(double)i,0);
FFT(a,1);FFT(b,1);
for(i=0;i<N;++i) c[i]=a[i]*b[i];
FFT(c,-1); for(i=0;i<n;++i) ans[i]=c[i].x;
memset(a,0,sizeof(a));memset(b,0,sizeof(b));
for(i=0;i<n;++i) a[i].prepare(qi[n-i-1],0);
for(i=1;i<n;++i) b[i].prepare(1.0/(double)i/(double)i,0);
FFT(a,1);FFT(b,1);
for(i=0;i<N;++i) c[i]=a[i]*b[i];
FFT(c,-1); for(i=0;i<n;++i) ans[i]-=c[n-i-1].x;
for(i=0;i<n;++i) printf("%.9f\n",ans[i]);
}