【BZOJ3527/ZJOI2014】力 FFT模板题

原题走这里

鉴于本机房似乎除了我以外所有人都能一眼看出来这是个卷积
然而我却盯了好久才看出来
所以大佬们可以直接退出
sa yo na la ~~

我们把原题中的式子左右分开来看，现在只考虑左边，即 $left(j)=\sum_{i<j}\frac{q_i}{(i-j)^2}$
经观察得知：
$left(1)=0$
$left(2)=\frac{q_1}{1^2}$
$left(3)=\frac{q_2}{1^2}+\frac{q_1}{2^2}$
$left(n)=\frac{q_{n-1}}{1^2}+\frac{q_{n-2}}{2^2}+...+\frac{q_1}{(n-1)^2}$
则向量left可以看作是$A(q_1,q_2,q_3,q_4,...,q_n)$与$B(0,1,\frac{1}{4},\frac{1}{9},...,\frac{1}{(n-1)^2})$的卷积
式子另一边推法也差不多，FFT即可

代码如下：

#include <bits/stdc++.h>
using namespace std;
int n,m;
const double pi=acos(-1);
complex<double> a[400010],q[400010],Q[400010],w[400010];
void init()
{
    for(int i=0;i<n;i++)
    {
        w[i]=complex<double>(cos(2.0*pi*i/n),sin(2.0*pi*i/n));
    }
}
void FFT(complex<double> *a)
{
    for(int i=0,j=0;i!=n;i++)
    {
        if(i>j)swap(a[i],a[j]);
        for(int k=n>>1;(j^=k)<k;k>>=1);
    }
    for(int i=2;i<=n;i<<=1)
    {
        int m=i>>1,l=n/i;
        for(int j=0;j<n;j+=i)
        {
            for(int k=0;k!=m;k++)
            {
                complex<double> t=a[j+m+k]*w[l*k];
                a[j+m+k]=a[j+k]-t;
                a[j+k]+=t;
            }
        }
    }
}
int main()
{
    ios::sync_with_stdio(0);
    cin>>n;
    m=n;
    for(int i=1;i!=n;i<<=1,n=max(n,i));
    n<<=1;
    for(int i=0;i<m;i++)
    {
        cin>>q[i];
        Q[m-i-1]=q[i];
        a[i]=(i?1.0/i/i:0);
    }
    init();
    FFT(q);
    FFT(Q);
    FFT(a);
    for(int i=0;i<n;i++)q[i]*=a[i],Q[i]*=a[i];
    reverse(w+1,w+n);
    FFT(q);
    FFT(Q);
    for(int i=0;i<m;i++)
    {
        printf("%.3lf\n",(q[i]-Q[m-i-1]).real()/n);
    }
    return 0;   
}