BZOJ[3527][Zjoi2014]力 FFT

传送门ber~

这题有毒好几次把$i,j$弄反

题目就是让求一个

$$E_j=\sum_{i=1}^{j-1} \frac{q_i}{(i-j)^2}-\sum_{i=j+1}^{n} \frac{q_i}{(i-j)^2}$$
令 $f(i)=\frac{1}{i^2}$, $g(i)=q_i$，那么
$$E_j=\sum_{i=1}^{j-1} g(i)*f(j-i)-\sum_{i=j+1}^{n} g(i)*f(j-i)$$
左面标准的卷积，FFT搞定
右面不是从0开始，FFT搞定不了
那咋整?
令 $g'(i)=q_{n-i}$
那么 $\sum \limits _{i=j+1}^{n} g(i)*f(j-i)$可以转化为
$$\sum \limits _{i=0}^{n-j-1} g(n-i-1)*f(n-i-j-1)$$
$$E_j=\sum \limits _{i=0}^{n-j-1} g'(i+1)*f(n-i-j-1)$$
转化一个卷积的形式，FFT搞定

当然也可以在开始的时候将多项式平移$n$位，结果取后$n$位

代码如下：

#include<algorithm>
#include<cstdio>
#include<cmath>
#define N 500050
using namespace std;
const double DFT=2.0,IDFT=-2.0;
const double pi=acos(-1);
struct Complex{
    double x,y;
    Complex(double _=0.0,double __=0.0):x(_),y(__){}
    Complex operator + (Complex b) {return Complex(x+b.x,y+b.y);}
    Complex operator - (Complex b) {return Complex(x-b.x,y-b.y);}
    Complex operator * (Complex b) {return Complex(x*b.x-y*b.y,x*b.y+b.x*y);}
}a[N],b[N];
int n,len;
int pos[N];
inline void FFT(Complex x[],double mode){
    for(int i=0;i<len;i++)
        if(i<pos[i])
            swap(x[i],x[pos[i]]);
    for(int i=2,mid=1;i<=len;i<<=1,mid<<=1){
        Complex wm(cos(2.0*pi/i),sin(mode*pi/i));
        for(int j=0;j<len;j+=i){
            int r=j+mid;
            Complex w(1,0);
            for(int k=j;k<r;k++){
                Complex y=x[k],z=w*x[k+mid];
                x[k]=y+z;x[k+mid]=y-z;
                w=w*wm;
            }
        }
    }
    if(mode==IDFT)
        for(int i=0;i<len;i++)
            x[i].x/=len;
    return;
}

int main(){
    scanf("%d",&n);
    for(int i=0;i<n;i++)
        scanf("%lf",&a[i].x);
    for(int i=0;i<n-1;i++)
        b[i].x=-1.0/(n-i-1)/(n-i-1);
    for(int i=n;i<n*2-1;i++)
        b[i].x=1.0/(i-n+1)/(i-n+1);
    for(len=1;len<n<<1;len<<=1);
    for(int i=0;i<len;i++){
        pos[i]=pos[i>>1]>>1;
        if(i&1) pos[i]|=len>>1;
    }
    FFT(a,DFT);FFT(b,DFT);
    for(int i=0;i<len;i++) a[i]=a[i]*b[i];
    FFT(a,IDFT);
    for(int i=n-1;i<n*2-1;i++) printf("%f\n",a[i].x);
return 0;
}