BZOJ_3527_[ZJOI2014]_力_(FFT+卷积)

描述

---

题面:

http://wenku.baidu.com/link?url=D2ORnA9xjgSxa2GlYLB7gGiYgBcXsy-Aw0kVYTjTE-iYhH1s7h8xXGmnaMwl32SYznVvzodyKZgHODl_ekzIFwEsO64ZOMIvQbsah-9kZiW

提交:

http://www.lydsy.com/JudgeOnline/problem.php?id=3527

给出n个数字(q1~qn),定义$$F_i=sum_{j<i}{q_iq_jover (i-j)^2}-sum_{j>i}{q_iq_jover (i-j)^2}$$

然后设$$E_i={F_iover q_i}$$

求  (E_i)

分析

---

我们先把式子化简一下$$E_i=sum_{j<i}{q_jover (i-j)^2}-sum_{j>i}{q_jover (i-j)^2}$$

然后我们令$$f[i]=q_i,g[i]={1over i^2}$$

然后发现左边好像卷积的形式$$c_i=sum_{j=0}^ia_jb_{i-j}$$

但是没有 (j=0) 和 (j=i) 的情况.没关系,我们令 (f[i]=0 , g[i]=0) .

这样的话原式[sum_{j=1}^{i-1}{q_jover (i-j)^2}]就和[sum_{j=0}^i{q_jover (i-j)^2}]相等了

这样左边就是[A_i=sum_{j=0}^if[j]g[i-j]]的卷积了

我们来看右边,右边也很有成为卷积的潜质啊,可是不满足(0le{j}le{i})啊.没关系,我们发现左边的式子和右边的式子正好相反,所以我们考虑"倒过来",

于是就有原式[sum_{j=i+1}^nf[j]g[i-j]]等价于[sum_{j=i}^nf[j]g[j-i]]又等价于[sum_{j=0}^{n-i}f[n-j]g[n-j-i]]

这个变化可以通过换元实现.我们可以设(f'[x]=f'[n-x]),这样的话右式就等于$$sum_{j=0}^{n-i}f'[j]g[n-i-j]$$

这样右边就是$$B_i=sum_{j=0}^if'[j]g[i-j]$$的卷积了

最后$$E_i=A_i+B_{n-i}$$

开心地FFT吧~

注意:

1.由于数组是从1开始的,预留了(f[0]与g[0])的位置,所以长度应该+1,所以len=2*n-1+1+1=2*n+1.

#include <bits/stdc++.h>
using namespace std;

const int maxn=263000;
const double pi=acos(-1.0);
int n,len;
int rev[maxn];
double f[maxn],f_[maxn],g[maxn],ans[maxn];
struct cp{
    double r,i;
    cp(double r=0,double i=0):r(r),i(i){}
    cp operator + (const cp &x) const { return cp(r+x.r,i+x.i); }
    cp operator - (const cp &x) const { return cp(r-x.r,i-x.i); }
    cp operator * (const cp &x) const { return cp(r*x.r-i*x.i,r*x.i+i*x.r); }
}a[maxn],b[maxn],c[maxn],A[maxn];
void brc(int &len){
    memset(rev,-1,sizeof rev);
    int k=1,l=0;
    while(k<len) k<<=1, l++;
    len=k;
    for(int i=1;i<len-1;i++){
        if(rev[i]!=-1) continue;
        int x=i,y=0,m=l;
        while(m--) y<<=1, y|=(x&1), x>>=1;
        rev[i]=y; rev[y]=i;
    }
}
void dft(cp *a,int n,int op){
    for(int i=1;i<n-1;i++) A[rev[i]]=a[i];
    for(int i=1;i<n-1;i++) a[i]=A[i];
    for(int m=2;m<=n;m<<=1){
        cp wn(cos(2.0*pi/m*op),sin(2.0*pi/m*op));
        for(int i=0;i<n;i+=m){
            cp w(1.0); int k=m>>1;
            for(int j=0;j<k;j++){
                cp u=a[i+j],t=w*a[i+j+k];
                a[i+j]=u+t;
                a[i+j+k]=u-t;
                w=w*wn;
            }
        }
    }
    if(op==-1)for(int i=0;i<n;i++) a[i].r/=n;
}
void fft(double *x,double *y,cp *a,cp* b,cp *c,int n){
    for(int i=0;i<len;i++) a[i]=cp(x[i]), b[i]=cp(y[i]);
    dft(a,n,1); dft(b,n,1);
    for(int i=0;i<n;i++) c[i]=a[i]*b[i];
    dft(c,n,-1);
}
int main(){
    scanf("%d",&n); len=2*n+1;
    brc(len);
    for(int i=1;i<=n;i++) scanf("%lf",&f[i]);
    for(int i=1;i<=n;i++) g[i]=1.0/i/i;
    for(int i=0;i<=n;i++) f_[i]=f[n-i];
    fft(f,g,a,b,a,len);
    for(int i=1;i<=n;i++) ans[i]=a[i].r;
    fft(f_,g,a,b,a,len);
    for(int i=1;i<=n;i++) ans[i]-=a[n-i].r;
    for(int i=1;i<=n;i++) printf("%.3lf
",ans[i]);
    return 0;
}