洛谷P3338：力（FFT）

传送门

解析

算是比较适合的FFT入门题了吧
一个重要的trick：
当函数无法表示成卷积时，可以把函数翻转过来
然后调一调就又是卷积了
一个重要的注意事项是FFT的lim一定是两多项式相乘结果多项式的项数！
即使后面的项根本没有用也一样
其他的比较简单了
注意下精度即可

代码

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define il inline
const int N=1e6+100;
const int M=150;
const int mod=998244353;
const double pi=acos(-1.0);
inline ll read(){
	ll x=0,f=1;char c=getchar();
	while(!isdigit(c)){if(c=='-') f=-1;c=getchar();}
	while(isdigit(c)){x=x*10+c-'0';c=getchar();}
	return x*f;
}
int n,m,lim,k;
struct node{
	double x,y;
	node(double a=0,double b=0){
		x=a;y=b;
	}
}A[N],B[N],C[N];
il node operator * (node a,node b){
	return (node){a.x*b.x-a.y*b.y,a.x*b.y+a.y*b.x};
}
il node operator + (node a,node b){
	return (node){a.x+b.x,a.y+b.y};
}
il node operator - (node a,node b){
	return (node){a.x-b.x,a.y-b.y};
}
int r[N];
il void fft(node *x,int lim,int flag){
	for(int i=0;i<=lim;i++){
		if(i<r[i]) swap(x[i],x[r[i]]);
	}
	for(int l=1;l<lim;l<<=1){
		node o(cos(pi/l),flag*sin(pi/l));
		for(int st=0;st<=lim;st+=l<<1){
			node t(1,0);
			for(int j=0;j<l;j++,t=t*o){
				node u=x[st+j],v=t*x[st+j+l];
				x[st+j]=u+v;
				x[st+j+l]=u-v;
			}
		}
	}
	return;
}
int main(){
	n=read();
	for(int i=1;i<=n;i++){
		scanf("%lf",&A[i].x);
		B[i].x=(double)(1.0/i/i);
		C[n-i].x=A[i].x;
	}
	int L=0,lim=1;
	while(lim<=n<<1){lim<<=1;L++;}
	for(int i=1;i<=lim;i++) r[i]=(r[i>>1]>>1)|((i&1)<<(L-1));
	fft(A,lim,1);fft(B,lim,1);fft(C,lim,1);
	for(int i=0;i<=lim;i++){
		A[i]=A[i]*B[i];C[i]=C[i]*B[i];
	}
	fft(A,lim,-1);fft(C,lim,-1);
	for(int i=1;i<=n;i++) printf("%lf\n",(A[i].x-C[n-i].x)/lim);
	return 0;
}
/*

*/