[bzoj3527][ZJOI2014]力

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=\sum_{i>j}^{}{qj \over (i-j)^2}-\sum_{i<j}^{}{qj \over (j-i)^2}$$
$$Ei=\sum_{j=1}^{i-1}{qj \over (i-j)^2}-\sum_{j=i+1}^{n}{qj \over (j-i)^2}$$
$$Ei=\sum_{j=1}^{i-1}{a[j]*b[i-j]}-\sum_{j=i+1}^{n}{a[j]*b[j-i]}$$
$$a[i]=qi,b[i]={1 \over i^2}$$
第一个式子已经化成了卷积的形式，对于第二个式子可以换一下元
$$\sum_{j=0}^{n-1-i}{a[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]);
}