1. 程式人生 > >多項式乘法

多項式乘法

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; }

多項式乘法