版权声明:转载请附带链接或评论 https://blog.csdn.net/corsica6/article/details/82781642
说在前面
发现
的代码里
卷积的姿势不太一样,于是学习了一发
。
三次 可以通过 优化成两次,卡常数必备 。orzmyy
共轭复数
两个实部相等,虚部互为相反数的复数互为共轭复数( )。将复数 的共轭复数记做 。
则有运算:
myyFFT
设当前卷积的两个多项式分别为
,设进行
的多项式为
。
复数
的实部记做
,虚部记做
。
则 。另设多项式 。
由 ,得:
在将
转化成点积表示后,由:
得到
的卷积表示为:
代码
#include<bits/stdc++.h>
#define RI register
#define db double
using namespace std;
const int N=5e5+10;
const db pi=acos(-1);
int n,m,L,len,rv[N];
struct cc{
db r,i;
cc(){r=i=0;};
cc(db r_,db i_){r=r_;i=i_;}
cc operator +(const cc&ky){return cc(r+ky.r,i+ky.i);}
cc operator -(const cc&ky){return cc(r-ky.r,i-ky.i);}
cc operator *(const cc&ky){return cc(r*ky.r-i*ky.i,r*ky.i+i*ky.r);}
cc operator /(const int&ky){return cc(r/(db)ky,i/(db)ky);}
inline cc conj(){return cc(r,-i);}
}a[N],b[N];
inline void FFT(cc *e,int ptr)
{
RI int i,j,k,t;cc ix,iy,bs,ori;
for(i=1;i<len;++i) if(i<rv[i]) swap(e[i],e[rv[i]]);
for(i=1;i<len;i<<=1){
ori=cc(cos(pi/i),ptr*sin(pi/i));
for(j=0;j<len;j+=(i<<1)){
bs=cc(1,0);
for(k=0;k<i;++k,bs=bs*ori){
ix=e[j+k];iy=bs*e[i+j+k];
e[j+k]=ix+iy;e[i+j+k]=ix-iy;
}
}
}
if(ptr==1) return;
for(i=0;i<len;++i) e[i]=e[i]/len;
}
int main(){
RI int i,j;
scanf("%d%d",&n,&m);
for(i=0;i<=n;++i) scanf("%lf",&a[i].r);
for(i=0;i<=m;++i) scanf("%lf",&a[i].i);
for(len=1,L=0;len<=n+m;len<<=1) L++;
for(i=1;i<len;++i) rv[i]=(rv[i>>1]>>1)|((i&1)<<(L-1));
FFT(a,1);
for(i=0;i<len;++i){
j=(len-i)&(len-1);
b[i]=(a[i]*a[i]-(a[j]*a[j]).conj())*cc(0,-0.25);
}
FFT(b,-1);
for(i=0;i<=n+m;++i)
printf("%d ",int(b[i].r+0.5));
return 0;
}