[BZOJ3527][Zjoi2014][FFT]力

题意

---

求每个$q_i$的$E_i=\sum_{j<i}\frac{q_j}{(i-j)^2}-\sum_{j>i}\frac{q_j}{(i-j)^2}$

---

裸的FFT….
第一发FFT……
令$f(i)=q_i$，$f'(i)=q_{n-i+1}$,$g(i)=\frac{1}{i^2}$
那么$E(i)=\sum_{j<i}f(i)g(j-i)-\sum_{j>i}f'(i)g(j-i)$
两个sigma都是卷积形式，套FFT

发现FFT还是背起来比较方便……

#include <cstdio>
#include <complex>
#include <cmath>
#include <algorithm>
#include <iostream>
#define N 500010

using namespace std;
typedef complex<double> E;

int an,n,m,L;
int R[N];
E Ans1[N],Ans2[N],A1[N],A2[N],B[N];
double q[N];

inline void FFT(E *a,int r){
    for(int i=0;i<n;i++) if(R[i]>i) swap(a[i],a[R[i]]);
    for(int i=1;i<n;i<<=1){
        E wn(cos(M_PI/i),r*sin(M_PI/i));
        for(int j=0;j<n;j+=(i<<1)){
            E w(1,0);
            for(int k=0;k<i;k++,w*=wn){
                E x=a[j+k],y=w*a[j+k+i];
                a[j+k]=x+y;a[j+k+i]=x-y;
            }
        }
    }
    if(r==-1) for(int i=0;i<n;i++) a[i]/=n;
}

int main(){
    freopen("force.in","r",stdin);
    freopen("force.out","w",stdout);
    scanf("%d",&an);an--;
    for(int i=0;i<=an;i++) scanf("%lf",&q[i]),A1[i]=A2[an-i]=q[i];
    for(int i=1;i<=an;i++) B[i]=1.0/i/i;
    for(n=1,m=an<<1;n<=m;n<<=1) L++;
    for(int i=0;i<n;i++) R[i]=(R[i>>1]>>1)|((i&1)<<(L-1));
    FFT(A1,1);FFT(A2,1);FFT(B,1);
    for(int i=0;i<n;i++) Ans1[i]=A1[i]*B[i],Ans2[i]=A2[i]*B[i];
    FFT(Ans1,-1);FFT(Ans2,-1);
    for(int i=0;i<=an;i++) printf("%.3lf\n",Ans1[i].real()-Ans2[an-i].real());
    return 0;
}