bzoj3527 [Zjoi2014]力（fft求卷积）

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

Icefox_zhx

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

阅读量227

点赞数

分类专栏： bzoj ---------数学--------- FFT/NTT/FWT/FMT/多项式

本文链接：https://blog.csdn.net/Icefox_zhx/article/details/79521687

版权

bzoj 同时被 3 个专栏收录

764 篇文章 1 订阅

订阅专栏

---------数学---------

209 篇文章 0 订阅

订阅专栏

FFT/NTT/FWT/FMT/多项式

16 篇文章 0 订阅

订阅专栏

求

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

$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}$
设

Cj=∑i=0j−1qi(j−i)2,Dj=∑i=j+1n−1qi(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}$ ,则

Ej=Cj−Dj $E_j=C_j-D_j$

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

C j = \sum i = 0 j - 1 q i ( j - i ) 2

$C_j=\sum\limits_{i=0}^{j-1}\frac{q_i}{(j-i)^2}$

C j = \sum i = 0 j a i * c j - i

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

再看D这一部分

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

$D_j=\sum\limits_{i=j+1}^{n-1}\frac{q_i}{(i-j)^2}$

D j = \sum i = j n - 1 a i * c i - j

$D_j=\sum\limits_{i=j}^{n-1}a_i*c_{i-j}$

D j = \sum i = 0 n - 1 - j a i + j * c i

$D_j=\sum\limits_{i=0}^{n-1-j}a_{i+j}*c_i$
我们令

bi=an−1−i $b_i=a_{n-1-i}$
则

D j = \sum i = 0 n - 1 - j b n - i - j - 1 * c i = P n - 1 - j

$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;
}