社区讨论
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 条回复,欢迎继续交流。
正在加载回复...