[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)\)

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

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

卷积用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;
}
上一篇:【荐1】Total Commander 7.57 个人使用设置 及 常用快捷键 备忘


下一篇:[bzoj3527] [洛谷P3338] [Zjoi2014]力