多项式乘法

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