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_j\over (i-j)^2}-\sum_{j>i}{q_iq_j\over (i-j)^2}$$

然后设$$E_i={F_i\over q_i}$$

求  \(E_i\)

分析

---

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

然后我们令$$f[i]=q_i,g[i]={1\over 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_j\over (i-j)^2}\]就和\[\sum_{j=0}^i{q_j\over (i-j)^2}\]相等了

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

我们来看右边,右边也很有成为卷积的潜质啊,可是不满足\(0\le{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=;
 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=,double i=):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,-,sizeof rev);
     int k=,l=;
     while(k<len) k<<=, l++;
     len=k;
     for(int i=;i<len-;i++){
         if(rev[i]!=-) continue;
         int x=i,y=,m=l;
         while(m--) y<<=, y|=(x&), x>>=;
         rev[i]=y; rev[y]=i;
     }
 }
 void dft(cp *a,int n,int op){
     for(int i=;i<n-;i++) A[rev[i]]=a[i];
     for(int i=;i<n-;i++) a[i]=A[i];
     for(int m=;m<=n;m<<=){
         cp wn(cos(2.0*pi/m*op),sin(2.0*pi/m*op));
         for(int i=;i<n;i+=m){
             cp w(1.0); int k=m>>;
             for(int j=;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==-)for(int i=;i<n;i++) a[i].r/=n;
 }
 void fft(double *x,double *y,cp *a,cp* b,cp *c,int n){
     for(int i=;i<len;i++) a[i]=cp(x[i]), b[i]=cp(y[i]);
     dft(a,n,); dft(b,n,);
     for(int i=;i<n;i++) c[i]=a[i]*b[i];
     dft(c,n,-);
 }
 int main(){
     scanf("%d",&n); len=*n+;
     brc(len);
     for(int i=;i<=n;i++) scanf("%lf",&f[i]);
     for(int i=;i<=n;i++) g[i]=1.0/i/i;
     for(int i=;i<=n;i++) f_[i]=f[n-i];
     fft(f,g,a,b,a,len);
     for(int i=;i<=n;i++) ans[i]=a[i].r;
     fft(f_,g,a,b,a,len);
     for(int i=;i<=n;i++) ans[i]-=a[n-i].r;
     for(int i=;i<=n;i++) printf("%.3lf\n",ans[i]);
     return ;
 }