多項式乘法
阿新 • • 發佈:2017-10-06
return main code clu %d wap 多項式 printf col
#include<cstdio> #include<complex> using namespace std; typedef complex<double> cd; const int maxn=1e6; const double pi=acos(-1); cd a[maxn+10],b[maxn+10]; int n,m,l,x,lgl=-1,r[maxn+10]; void fft(cd x[],int d){ for(int i=0;i<l;++i) if(i<r[i]) swap(x[i],x[r[i]]);for(int i=1;i<l;i*=2){ cd wn(cos(pi/i),d*sin(pi/i)); for(int j=0;j<l;j+=i*2){ cd w(1,0); for(int k=0;k<i;++k,w*=wn){ cd a=x[j+k],b=x[i+j+k]*w; x[j+k]=a+b; x[i+j+k]=a-b; } } } } int main(){ scanf("%d%d",&n,&m); for(int i=0;i<=n;++i){ scanf("%d",&x); a[i]=x; } for(int i=0;i<=m;++i){ scanf("%d",&x); b[i]=x; } for(l=1;l<=n+m;l*=2) ++lgl; for(int i=0;i<l;++i) r[i]=(r[i/2]/2)|(i%2)<<lgl; fft(a,1); fft(b,1); for(int i=0;i<l;++i) a[i]*=b[i]; fft(a,-1); for(int i=0;i<=n+m;++i) printf("%d ",int(a[i].real()/l+0.5)); return 0; }
多項式乘法