[Luogu P3338] [ZJOI2014]力 (数论 FFT 卷积)

题面

传送门:


Solution

写到脑壳疼,我好菜啊

我们来颓柿子吧
\(F_j=\sum_{i<j}\frac{q_i*q_j}{(i-j)^2}-\sum_{i>j}\frac{q_i*q_j}{(i-j)^2}\)

\(q_j\)\(i\)没有半毛钱关系,提到外面去
\(F_j=q_j*\sum_{i<j}\frac{q_i}{(i-j)^2}-q_j*\sum_{i>j}\frac{q_i}{(i-j)^2}\)

左右同时除以\(q_j\)
\(E_j=\sum_{i=1}^{j-1}\frac{q_i}{(i-j)^2}-\sum_{i=j+1}^{n}\frac{q_i}{(i-j)^2}\)

我们设\(f(i)=q(i),g(i)=\frac{1}{i^2}\),有
\(E_j=\sum_{i=1}^{j-1}f(i)*g(i-j)-\sum_{i=j+1}^{n}f(i)*g(i-j)\)

因为\(g(i)\)是个偶函数,因此有:
\(E_j=\sum_{i=1}^{j-1}f(i)*g(j-i)-\sum_{i=j+1}^{n}f(i)*g(i-j)\)

这时候,我们显然可以发现左边那个式子是个卷积,右边的这样一波化简就也变成了卷积形式:

卷积用FFT快速计算即可

时间复杂度\(O(nlogn)\)


Code

//Luogu P3338 [ZJOI2014]力
//Jan,18th,2019
//FFT加速卷积
#include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>
#include<complex>
using namespace std;
typedef complex <double> cp;
const double PI=acos(-1);
const int M=100000+100;
const int N=8*M;
inline cp omega(int K,int n)
{
    return cp(cos(2*PI*K/n),sin(2*PI*K/n));
}
void FFT(cp a[],int n,bool type)
{
    static int len=0,num=n-1,t[N];
    while(num!=0) len++,num/=2;
    for(int i=0,j;i<=n;i++)
    {
        for(j=0,num=i;j<len;j++)
            t[j]=num%2,num/=2;
        reverse(t,t+len);
        for(j=0,num=0;j<len;j++)
            num+=t[j]*(1<<j);
        if(i<num) swap(a[i],a[num]);
    }
    for(int l=2;l<=n;l*=2)
    {
        int m=l/2;
        cp x0=omega(1,l);
        if(type==true) x0=conj(x0);
        for(int j=0;j<n;j+=l)
        {
            cp x=cp(1,0);
            for(int k=0;k<m;k++,x*=x0)
            {
                cp temp=x*a[j+k+m];
                a[j+k+m]=a[j+k]-temp;
                a[j+k]=a[j+k]+temp;
            }
        }
    }
}
int n,m;
double q[N];
cp f[N],g[N],f2[N];
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
        scanf("%lf",&q[i]);
        
    for(int i=1;i<=n;i++)
        g[i]=(1.0/i/i);
    m=1;
    while(m<2*n) m*=2;
    for(int i=1;i<m;i++)
        f[i]=q[i],f2[i]=q[i];
    
    FFT(g,m,false);
    FFT(f,m,false);
    reverse(f2+1,f2+n+1);
    FFT(f2,m,false);
    for(int i=0;i<m;i++)
        f[i]*=g[i],f2[i]*=g[i];
    FFT(f,m,true);
    FFT(f2,m,true);
    
    for(int i=1;i<=n;i++)
        printf("%lf\n",(f[i].real()-f2[n-i+1].real())/m);
    return 0;
}
posted @ 2019-01-18 16:12 GoldenPotato 阅读(...) 评论(...) 编辑 收藏