BZOJ[3527][Zjoi2014]力 FFT

最新推荐文章于 2018-12-10 10:49:45 发布

Duan2baka

最新推荐文章于 2018-12-10 10:49:45 发布

阅读量258

点赞数

分类专栏： BZOJ 各省省选 FFT/NTT

本文链接：https://blog.csdn.net/WADuan2/article/details/79478497

版权

BZOJ 同时被 3 个专栏收录

118 篇文章 0 订阅

订阅专栏

各省省选

33 篇文章 0 订阅

订阅专栏

FFT/NTT

8 篇文章 0 订阅

订阅专栏

传送门ber~

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

题目就是让求一个

E j = \sum i = 1 j - 1 q i ( i - j ) 2 - \sum i = j + 1 n q i ( i - j ) 2

$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)=1i2 f ( i ) = 1 i 2 $f(i)=\frac{1}{i^2}$ ,

g(i)=qi g ( i ) = q i $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)

$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)=qn−i g ′ ( i ) = q n − i $g'(i)=q_{n-i}$
那么

∑i=j+1ng(i)∗f(j−i) ∑ i = j + 1 n g ( i ) ∗ f ( j − i ) $\sum \limits _{i=j+1}^{n} g(i)*f(j-i)$ 可以转化为

\sum i = 0 n - j - 1 g (n - i - 1) * f (n - i - j - 1)

$\sum \limits _{i=0}^{n-j-1} g(n-i-1)*f(n-i-j-1)$

E j = \sum i = 0 n - j - 1 g' (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;
}