社区讨论

FFT求调

P3338[ZJOI2014] 力参与者 2已保存回复 1

讨论操作

快速查看讨论及其快照的属性,并进行相关操作。

当前回复
1 条
当前快照
1 份
快照标识符
@lue5a6b1
此快照首次捕获于
2024/03/30 21:45
2 年前
此快照最后确认于
2024/03/31 01:15
2 年前
查看原帖
CPP
#include<bits/stdc++.h>
using namespace std;
const int maxn=262144+10;
const long double PI=acos(-1);
int Bit[maxn],n,N;
complex<long double>A[maxn],B[maxn],C[maxn];
long double a[maxn];
void FFT(complex<long double> *H,int op){
	for(int i=0;i<N;i++)
		if(i<Bit[i]) swap(H[i],H[Bit[i]]);
	for(int len=2;len<=N;len<<=1){
		int child=len/2;
		complex<long double> o_fir={cos(2*PI/len),op*sin(2*PI/len)};
		for(int L=0;L<N;L+=len){
			complex<long double> o_now(1,0);
			for(int i=L;i<L+child;i++){
				complex<long double> tmp=o_now*H[i+child];
				H[i+child]=H[i]-tmp; 
				H[i]=H[i]+tmp;
				o_now=o_now*o_fir;
			}
		}
	}
}
int main(){
	scanf("%d",&n);
	for(int i=1;i<=n;i++) scanf("%Lf",&a[i]);
	for(int i=1;i<=n;i++)
		A[i]={a[i],0},B[i]={a[n-i],0},C[i]={(long double)((long double)1.0/i/i),0};
	for(N=1;N<=n*2;N<<=1);
	for(int i=0;i<N;i++) Bit[i]=(Bit[i>>1]>>1)|((i&1)?(N>>1):0);
	FFT(B,1),FFT(A,1),FFT(C,1);
	for(int i=0;i<N;i++) A[i]=A[i]*C[i],B[i]=B[i]*C[i];
	FFT(A,-1),FFT(B,-1);
	for(int i=0;i<N;i++) A[i]=A[i].real()/N,B[i]=B[i].real()/N;
	for(int i=1;i<=n;i++) printf("%.3Lf\n",(A[i].real()-B[n-i].real()));
	return 0;
} 
过不去样例。
DFT对A的结果是正确的,对B的结果是错误的。
感觉是DFT细节出了错误。
但是这份FFT模板能过模板题。

回复

1 条回复,欢迎继续交流。

正在加载回复...