【UOJ 34】 #34. 多项式乘法 （FFT）

【分析】

　　这个只是用来放模板。。【其实我还没完全懂的。。

迭代 代替 递归：

 #include<cstdio>
 #include<cstdlib>
 #include<cstring>
 #include<iostream>
 #include<algorithm>
 #include<cmath>
 #include<complex>
 #define Maxn 262145
 #define pi acos(-1)
 using namespace std;

 struct P
 {
     double x,y;
     P() {x=y=;}
     P(double x,double y):x(x),y(y){}
     friend P operator + (P x,P y) {return P(x.x+y.x,x.y+y.y);}
     friend P operator - (P x,P y) {return P(x.x-y.x,x.y-y.y);}
     friend P operator * (P x,P y) {return P(x.x*y.x-x.y*y.y,x.x*y.y+x.y*y.x);}
 }a[Maxn],b[Maxn];

 int nn;
 int R[Maxn];
 void fft(P *a,int f)
 {
     for(int i=;i<nn;i++) if(i<R[i]) swap(a[i],a[R[i]]);
     for(int i=;i<nn;i<<=)
     {
         P wn(cos(pi/i),f*sin(pi/i));
         for(int j=;j<nn;j+=i<<)
         {
             P w(,);
             for(int k=;k<i;k++,w=w*wn)
             {
                 P x=a[j+k],y=w*a[j+k+i];
                 a[j+k]=x+y;a[j+k+i]=x-y;
             }
         }
     }
 }
 int main()
 {
     freopen("a.in","r",stdin);
     freopen("b.out","w",stdout);
     int n,m;
     scanf("%d%d",&n,&m);
     for(int i=;i<=n;i++) scanf("%lf",&a[i].x);
     for(int i=;i<=m;i++) scanf("%lf",&b[i].x);
     int ll=;nn=;
     while(nn<=n+m) ll++,nn<<=;
     for(int i=;i<nn;i++) R[i]=(R[i>>]>>)|((i&)<<(ll-));
     fft(a,);fft(b,);
     for(int i=;i<=nn;i++) a[i]=a[i]*b[i];
     fft(a,-);
     for(int i=;i<=n+m;i++) printf("%d ",(int)(a[i].x/nn+0.5));
     return ;
 }

2017-04-13 16:43:54