洛谷 P3338 [ZJOI2014]力

咕咕咕~~~
昨天刚学会FFT,今天早上本来不想做的,结果去看莫反发现看不懂,于是又回归了FFT……
然后发现我除了两道模板题其他题都不会QAQ
(I am so weak)

这道题我采用了抄题解的学习方式
看题解的推导看的一脸懵逼
后来又去看LJY大佬的blog,才发现题解有个地方写错了
最后又看了看绿鸟chen_zhe的题解才稍微明白一点
目前虽然AC了但是还是觉得要打篇笔记理顺一下

声明:本次推导使用从0开始的下标(虽然我不习惯)
首先,定义 g x = 1 x 2 g_x=\frac{1}{x^2} gx=x21

E j = F j q j = ∑ i < j q i q j ( i − j ) 2 − ∑ i > j q i q j ( i − j ) 2 q j = ∑ i < j q i g i − j − ∑ i > j q i g i − j = ∑ i = 0 j − 1 q i g j − i − ∑ i = j + 1 n − 1 q i g i − j \begin{aligned} E_j=&\frac{F_j}{q_j}\\ =& \large\frac{ {\sum\limits_{ij}\frac{q_iq_j}{(i-j)^2}} }{q_j}\\ =& {\sum\limits_{ij}q_ig_{i-j}} \\ =& {\sum\limits_{i=0}^{j-1}q_ig_{j-i}}- {\sum\limits_{i=j+1}^{n-1}q_ig_{i-j}} \end{aligned} Ej====qjFjqji<j(ij)2qiqji>j(ij)2qiqji<jqigiji>jqigiji=0j1qigjii=j+1n1qigij

我们再想一想,多项式相乘每一项系数的计算公式是什么?
F k = ∑ i = 0 k A i B k − i \large F_k=\sum\limits_{i=0}^{k}A_iB_{k-i} Fk=i=0kAiBki
减号前面那一项很像诶?
但是可惜上界是 j − 1 j-1 j1
其实解决这个问题也不难,只需要把 A A A向高次方向移一位就好了
移完就可以直接FFT算咯
后面的呢?
恰好反过来了诶!
那我们开一个 p p p,使 p i = q n − i p_i=q_{n-i} pi=qni就好了
最终得到:
E j = ∑ i = 0 j − 1 q i g j − i − ∑ i = j + 1 n − 1 p n − i g i − j \large E_j={\sum\limits_{i=0}^{j-1}q_ig_{j-i}}- {\sum\limits_{i=j+1}^{n-1}p_{n-i}g_{i-j}} Ej=i=0j1qigjii=j+1n1pnigij
A = q g , B = p g A=qg,B=pg A=qg,B=pg
E j = A j − ∑ i = j + 1 n − 1 p n − i g i − j \large E_j={A_j}- {\sum\limits_{i=j+1}^{n-1}p_{n-i}g_{i-j}} Ej=Aji=j+1n1pnigij

AC代码送上:

#include
#include
#include
#include
#define cp complex
using namespace std;
const double pi=acos(-1);
int R[262145];
void fft(cp *arr,int len,int inv)
{
	for(int n=2;n<=len;n<<=1)
	{
		const cp ur(cos(2*pi/n),sin(2*pi*inv/n));
		const int mid=n>>1;
		for(cp *a=arr;a<arr+len-1;a+=n)
		{
			cp r(1,0),t;
			for(int i=0;i<mid;i++,r*=ur)
			{
				t=a[mid+i]*r;
				a[mid+i]=a[i]-t;
				a[i]=a[i]+t;
			}
		}
	}
}
void dft(cp *a,int n,bool inv=0)
{
	for(int i=0;i<n;i++)
		if(i<R[i])
			swap(a[i],a[R[i]]);
	fft(a,n,inv?-1:1);
	if(inv)
		for(int i=0;i<n;i++)
			a[i].real(a[i].real()/n);
}
cp a[262145],b[262145],c[262145];
int main()
{
	int n,m;
	scanf("%d",&n);
	for(int i=1;i<=n;i++)\\从1开始,自动高移
		scanf("%lf",&a[i]),
		c[i]=1.0/(1ll*i*i),
		b[n-i+1]=a[i];
	m=n<<1;for(n=1;n<m;n<<=1);m>>=1;
	for(int i=1;i<n;i++)
		R[i]=R[i>>1]>>1|(i&1?n>>1:0);
	dft(a,n);dft(b,n);dft(c,n);
	for(int i=1;i<=n;i++)
		a[i]*=c[i];
	for(int i=1;i<=n;i++)
		b[i]*=c[i];
	dft(a,n,1);dft(b,n,1);
	for(int i=1;i<=m;i++)
		printf("%.10lf\n",a[i].real()-b[m-i+1].real());
	return 0;
}

你可能感兴趣的:(做题笔记,洛谷,FFT)