bzoj3527 [Zjoi2014]力（fft求卷积）

求

$$E_j=\sum\limits_{i=0}^{j-1}\frac{q_i}{(j-i)^2}-\sum\limits_{i=j+1}^{n-1}\frac{q_i}{(i-j)^2}$$
设 $C_j=\sum\limits_{i=0}^{j-1}\frac{q_i}{(j-i)^2},D_j=\sum\limits_{i=j+1}^{n-1}\frac{q_i}{(i-j)^2}$,则 $E_j=C_j-D_j$

令$a_i=q_i,c_i=\frac{1}{i^2}(c_0=0)$
则

$$C_j=\sum\limits_{i=0}^{j-1}\frac{q_i}{(j-i)^2}$$
$$C_j=\sum\limits_{i=0}^{j}a_i*c_{j-i}$$ 就是个卷积可以fft求。

再看D这一部分

$$D_j=\sum\limits_{i=j+1}^{n-1}\frac{q_i}{(i-j)^2}$$
$$D_j=\sum\limits_{i=j}^{n-1}a_i*c_{i-j}$$
$$D_j=\sum\limits_{i=0}^{n-1-j}a_{i+j}*c_i$$
我们令 $b_i=a_{n-1-i}$
则
$$D_j=\sum\limits_{i=0}^{n-1-j}b_{n-i-j-1}*c_i=P_{n-1-j}$$
也是个卷积，倒过来就好了。

于是就是两个裸卷积。

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <complex>
using namespace std;
#define N 270000
#define pi acos(-1)
typedef complex<double> E;
int n,m,L=0,R[N];
E a[N],b[N],c[N];
inline void fft(E *a,int f){
    for(int i=0;i<n;++i) if(i<R[i]) swap(a[i],a[R[i]]);
    for(int i=1;i<n;i<<=1){
        E wn(cos(pi/i),f*sin(pi/i));
        for(int p=i<<1,j=0;j<n;j+=p){
            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(f==-1) for(int i=0;i<n;++i) a[i]/=n;
}
int main(){
//  freopen("a.in","r",stdin);
    scanf("%d",&n);
    for(int i=0;i<n;++i) scanf("%lf",&a[i]),b[n-1-i]=a[i];
    for(int i=1;i<n;++i) c[i]=1.0/i/i;
    m=n+n;for(n=1;n<=m;n<<=1) ++L;
    for(int i=0;i<n;++i) R[i]=(R[i>>1]>>1)|(i&1)<<L-1;
    fft(a,1);fft(c,1);fft(b,1);
    for(int i=0;i<n;++i) a[i]*=c[i],b[i]*=c[i];
    fft(a,-1);fft(b,-1);m>>=1;
    for(int i=0;i<m;++i) printf("%.3lf\n",a[i].real()-b[m-1-i].real());
    return 0;
}