bzoj3527: [Zjoi2014]力

链接

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

题解

　　我终于也能自己独立做出fft题了，还是1A，好开心。
　　这个题，你观察下分母，发现那个东西不就是距离的平方嘛。所以构造一个多项式，第i个数是$\frac{1}{i^2}(i>0)$。
　　这样求下$q$和这个多项式的卷积，就能得到光包含式子中第一项的一个多项式。
　　再把$q$串翻过来，在和自己构造的那个多项式求下卷积，最后把这两个多项式合并下就好了。

代码

//FFT
#include <cstdio>
#include <algorithm>
#include <cmath>
#define maxn (262144+100)
#define real long double
using namespace std;
int n, up, R[maxn];
real ans[maxn], q[maxn];
const real pi=acos(-1);
struct cp
{
    real x, y;
    cp(real x, real y):x(x),y(y){}
    cp(){}
}a[maxn], b[maxn], t[maxn];
inline cp operator+(cp a, cp b){return cp(a.x+b.x,a.y+b.y);}
inline cp operator-(cp a, cp b){return cp(a.x-b.x,a.y-b.y);}
inline cp operator*(cp a, cp b){return cp(a.x*b.x-a.y*b.y,a.x*b.y+a.y*b.x);}
void fft(cp *a, int n, int opt)
{
    int i, j, k, p; cp wn, w, x, y;
    for(i=0;i<n;i++)if(i>R[i])swap(a[i],a[R[i]]);
    for(i=1;i<n;i<<=1)
    {
        wn=cp(cos(pi/i),opt*sin(pi/i));
        for(p=i<<1,j=0;j<n;j+=p)
            for(w=cp(1,0),k=0;k<i;k++,w=w*wn)
            {
                x=a[j+k], y=a[j+k+i]*w;
                a[j+k]=x+y, a[j+k+i]=x-y;
            }
    }
    if(opt==-1)for(i=0;i<n;i++)a[i].x/=n;
}
void init()
{
    int i, L=0;
    scanf("%d",&n),n--;
    for(i=0;i<=n;i++)scanf("%Lf",&q[i]);
    for(up=1;up<=n<<1;up<<=1)L++;
    for(i=0;i<up;i++)R[i]=(R[i>>1]>>1)|((i&1)<<(L-1));
}
inline void juan(cp *a, cp *b, cp *c)
{
    fft(a,up,1), fft(b,up,1);
    for(int i=0;i<up;i++)c[i]=a[i]*b[i];
    fft(c,up,-1);
}
void work()
{
    int i;
    for(i=0;i<=n;i++)a[i]=cp(q[i],0);
    for(i=1;i<=n;i++)b[i].x=(real)1.0/((real)i*i);
    juan(a,b,t);for(i=0;i<up;i++)ans[i]=t[i].x;
    for(i=0;i<up;i++)a[i]=b[i]=cp(0,0);
    for(i=0;i<=n;i++)a[i]=cp(q[n-i],0);
    for(i=1;i<=n;i++)b[i].x=(real)1.0/((real)i*i);
    juan(a,b,t);for(i=0;i<=n;i++)ans[n-i]=ans[n-i]-t[i].x;
    for(i=0;i<=n;i++)printf("%Lf\n",ans[i]);
}
int main()
{
    init();
    work();
    return 0;
}