概述
例题:POJ2821
解析:
首先我们知道一般的,最常用的FFT求的就是在 % 2 n %2^n %2n意义下的循环卷积。
换句话说,长度为 n n n的 D F T DFT DFT求的就是在长度 % n %n %n下的循环卷积。
现在考虑长度不为 2 2 2的整数次幂的时候我们怎么办。
令长度为 n n n,还是参考 D F T DFT DFT的式子: A k = ∑ i = 0 n − 1 a i ω n i k A_k=sumlimits_{i=0}^{n-1}a_iomega_{n}^{ik} Ak=i=0∑n−1aiωnik
由于 2 i k = i 2 + k 2 − ( i − k ) 2 2ik=i^2+k^2-(i-k)^2 2ik=i2+k2−(i−k)2,于是我们可以把单位根换成 2 n 2n 2n次再来考虑。
于是 A k = ω 2 n k 2 ∑ i = 0 n − 1 a i ω 2 n i 2 ⋅ ω 2 n − ( i − k ) 2 A_k=omega_{2n}^{k^2}sum_{i=0}^{n-1}a_iomega_{2n}^{i^2}cdotomega_{2n}^{-(i-k)^2} Ak=ω2nk2i=0∑n−1aiω2ni2⋅ω2n−(i−k)2
发现 Σ Sigma Σ后面是一个卷积,于是我们考虑用卷积实现 D F T DFT DFT就行了。
构造两个序列,长度分别为 n n n和 2 n 2n 2n,具体实现可以参考代码里面。
代码:
#include<cmath>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
#define re register
#define cs const
cs double PI=acos(-1);
struct Complex{
double x,y;
Complex(){}
Complex(cs double &_x,cs double &_y):x(_x),y(_y){}
Complex conj()cs{return Complex(x,-y);}
friend Complex operator+(cs Complex &a,cs Complex &b){return Complex(a.x+b.x,a.y+b.y);}
friend Complex operator-(cs Complex &a,cs Complex &b){return Complex(a.x-b.x,a.y-b.y);}
friend Complex operator*(cs Complex &a,cs Complex &b){return Complex(a.x*b.x-a.y*b.y,a.x*b.y+a.y*b.x);}
friend Complex operator/(cs Complex &a,cs double &b){return Complex(a.x/b,a.y/b);}
friend Complex operator/(cs Complex &a,cs Complex &b){return a*b.conj()/(b.x*b.x+b.y*b.y);}
};
cs int N=1<<17|1;
int r[N<<2|1];
Complex w[N<<2|1];
inline void FFT(Complex *A,int len,int typ){
if(typ==-1)std::reverse(A+1,A+len);
for(int re i=0;i<len;++i)if(i<r[i])std::swap(A[i],A[r[i]]);
for(int re i=1;i<len;i<<=1)
for(int re j=0;j<len;j+=i<<1)
for(int re k=0;k<i;++k){
static Complex x,y;
x=A[j+k],y=A[j+k+i]*w[len/i/2*k];
A[j+k]=x+y;
A[j+k+i]=x-y;
}
if(typ==-1)for(int re i=0;i<len;++i)A[i]=A[i]/len;
}
inline void init_rev(int len){
static int last;if(last==len)return ;last=len;
for(int re i=0;i<len;++i)
r[i]=r[i>>1]>>1|((i&1)*(len>>1)),
w[i]=Complex(cos(2*PI*i/len),sin(2*PI*i/len));
}
inline void BS(Complex *a,int l,int typ){
static Complex A[N<<2|1],B[N<<2|1];
int l2=l<<1;
for(int re i=0;i<l;++i)A[i]=a[i]*Complex(cos(PI*i*i/l),typ*sin(PI*i*i/l));
for(int re i=0;i<l2;++i)B[i]=Complex(cos(PI*(i-l)*(i-l)/l),-typ*sin(PI*(i-l)*(i-l)/l));
int len=1;while(len<l*3)len<<=1;
for(int re i=l;i<len;++i)A[i]=Complex(0,0);
for(int re i=l2;i<len;++i)B[i]=Complex(0,0);
init_rev(len);
FFT(A,len,1),FFT(B,len,1);
for(int re i=0;i<len;++i)A[i]=A[i]*B[i];
FFT(A,len,-1);
for(int re i=0;i<l;++i)a[i]=A[i+l]*Complex(cos(PI*i*i/l),typ*sin(PI*i*i/l));
if(typ==-1)for(int re i=0;i<l;++i)a[i]=a[i]/l;
}
int n;
Complex a[N],b[N],c[N];
signed main(){
scanf("%d",&n);
for(int re i=0;i<n;++i)scanf("%lf",&b[i].x);
for(int re i=0;i<n;++i)scanf("%lf",&c[i].x);
BS(b,n,1),BS(c,n,1);
for(int re i=0;i<n;++i)a[i]=c[i]/b[i];
BS(a,n,-1);
for(int re i=0;i<n;++i)printf("%.4fn",a[i].x);
return 0;
}
最后
以上就是追寻吐司为你收集整理的【模板】Bluestein's algorithm 求循环卷积例题:POJ2821解析:的全部内容,希望文章能够帮你解决【模板】Bluestein's algorithm 求循环卷积例题:POJ2821解析:所遇到的程序开发问题。
如果觉得靠谱客网站的内容还不错,欢迎将靠谱客网站推荐给程序员好友。
发表评论 取消回复