BZOJ 3527 [Zjoi2014]力 FFT

题意:链接

方法: FFT

解析:

这题比那什么 万径人踪灭 和谐多了。

求所有的$E_i=\frac{F_i}{q_i}$

首先既然它除掉了$q_i$所以我们可以直接把$q_i$干掉。

然后呢观察整个的形式

如果对于一个序列

${a_1,a_2,a_3,a_4,a_5}$

代入公式则可以得到

$E_1=-\frac{a_2}{1^2}-\frac{a_3}{2^2}-\frac{a_4}{3^2}-\frac{a_5}{4^2}$

$E_2=\frac{a_1}{1^2}-\frac{a_3}{2^2}-\frac{a_4}{3^2}-\frac{a_5}{4^2}$

$E_3=\frac{a_1}{2^2}+\frac{a_2}{1^2}-\frac{a_4}{1^2}-\frac{a_5}{2^2}$

$E_4=\frac{a_1}{3^2}+\frac{a_2}{2^2}+\frac{a_3}{1^2}-\frac{a_5}{1^2}$

$E_5=\frac{a_1}{4^2}+\frac{a_2}{3^2}+\frac{a_3}{2^2}+\frac{a_4}{1^2}$

发现分子与分母有规律，于是考虑构造函数。

$a_1*x^0+a_2*x^1+a_3*x^2+a_4*x^3+a_5*x^4+0*x^5+0*x^6+0*x^7+0*x^8$

与

$-\frac{1}{4^2}*x^0-\frac{1}{3^2}*x^1-\frac{1}{2^2}*x^2-\frac{1}{1^2}*x^3+0*x^4+\frac{1}{1^2}*x^5+\frac{1}{2^2}*x^6+\frac{1}{3^2}*x^7+\frac{1}{4^2}*x^8$

将两个多项式相乘即可，取次数为n~2*n中的系数即可。

代码:

#include <cmath>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define N 524288
#define pi acos(-1)
using namespace std;
int n;
int rev[N];
double c[N];
struct complex
{
    double r,i;
    complex(double x=0.0,double y=0.0){r=x,i=y;}
    complex operator + (const complex a)
    {return complex(a.r+r,a.i+i);}
    complex operator - (const complex a) 
    {return complex(r-a.r,i-a.i);}
    complex operator * (const complex a)
    {return complex(r*a.r-i*a.i,r*a.i+i*a.r);}
}a[N],b[N];
void FFT(complex *a,int f)
{
    for(int i=0;i<n;i++)if(i<rev[i])swap(a[i],a[rev[i]]);
    for(int h=2;h<=n;h<<=1)
    {
        complex wn(cos(2*pi*f/h),sin(2*pi*f/h));
        for(int i=0;i<n;i+=h)
        {
            complex w(1,0);
            for(int j=0;j<(h>>1);j++,w=w*wn)
            {
                complex t=a[i+j+(h>>1)]*w;
                a[i+j+(h>>1)]=a[i+j]-t;
                a[i+j]=a[i+j]+t;
            }
        }
    }
    if(f==-1)for(int i=0;i<n;i++)a[i].r/=n;
}
int main()
{
    scanf("%d",&n);n--;
    for(int i=0;i<=n;i++)scanf("%lf",&a[i].r);
    for(int i=0;i<n;i++)
        b[i].r=(-1.0)/((double)(n-i)*(double)(n-i));
    for(int i=n+1;i<=2*n;i++)
        b[i].r=-b[2*n-i].r;
    int m=4*n,L=0,nn=n;
    for(n=1;n<=m;n<<=1)L++;
    for(int i=0;i<n;i++)rev[i]=(rev[i>>1]>>1)|((i&1)<<(L-1));
    FFT(a,1),FFT(b,1);
    for(int i=0;i<=n;i++)a[i]=a[i]*b[i];
    FFT(a,-1);
    for(int i=0;i<=m;i++)c[i]=a[i].r;
    for(int i=nn;i<=2*nn;i++)printf("%lf\n",c[i]);
}