【UOJ 34】多项式乘法

FFT模板

 

迭代的还没会 先写了个递归的

define的PI 我也是神了！！  少上一位就会WA

模板看的hzwer的

 

因为 PI较短【...】所以递归的跑的和迭代的一样快 23333333

 

 还差的几点：

1. acos要用

2.complex要自己写（yts1999大爷说会被卡）

3. 要改成迭代的

 

 

 1 #include <cstdio>
 2 #include <cmath>
 3 #include <complex>
 4 #include <iostream>
 5  using  namespace std;
 6  #define PI 3.14159265
 7  // const double PI =acos(-1);
 8  int n,m;
 9 complex< double> a[ 300000+ 1],b[ 300000+ 1];
10 
11  void FFT(complex< double> x[], int lenth, int type)
12 {
13      if(lenth== 1)  return ;
14     complex< double> p[lenth/ 2],q[lenth/ 2];
15      for( int i= 0;i<lenth;i++)
16      if(i& 1) p[i/ 2]=x[i];
17      else q[i/ 2]=x[i];
18     FFT(q,lenth/ 2,type);
19     FFT(p,lenth/ 2,type);
20     complex < double> wn(cos( 2*PI/lenth),sin(type* 2*PI/lenth)),t,w( 1, 0);
21      for( int i= 0;i<lenth/ 2;i++)
22     {
23         t=w*p[i];
24         x[i]=q[i]+t;
25         x[i+lenth/ 2]=q[i]-t;
26         w*=wn;
27     }
28 }
29  int main()
30 {
31     scanf( " %d %d ",&n,&m); int tmp;    
32 
33      for( int i= 0;i<=n;i++) scanf( " %d ",&tmp),a[i]=tmp;
34      for( int i= 0;i<=m;i++) scanf( " %d ",&tmp),b[i]=tmp;
35     m+=n;m++;
36     n= 1;
37      for(;n<m;n*= 2);
38     FFT(a,n, 1);
39     FFT(b,n, 1);
40      for( int i= 0;i<n;i++) a[i]*=b[i];
41     FFT(a,n,- 1);
42      for( int i= 0;i<m;i++) printf( " %d  ",( int)(a[i].real()/n+ 0.5));
43      return  0;
44 }

 

 

Upd:1.24 WC RP++a啊

 

#include <cstdio>
#include <iostream>
#include <complex>
#include <cmath>
using  namespace std;
const  double PI=acos(- 1);
// #define PI 3.14159265
#define C complex<double> 
C a[ 300000],b[ 300000];
int rev[ 300000];
int n,m;
void FFT(C x[], int type)
{
     for( int i= 0;i<n;i++)  if(rev[i]<i) swap(x[i],x[rev[i]]);
//     for(int i=0;i<n;i++) cout<<x[i]<<' ';cout<<endl;
     for( int i= 2;i<=n;i*= 2)
    {
        C wn(cos( 2*PI/i),type*sin( 2*PI/i));
     //     cout<<wn<<endl;
         for( int j= 0;j<n;j+=i)
        {
            C w( 1, 0),t,tmp;
             for( int k= 0;k<i/ 2;k++)
            {
                tmp=x[j+k];
                t=x[j+k+i/ 2]*w;
                x[j+k]=tmp+t;
                x[j+k+i/ 2]=tmp-t;
                w=w*wn;
            }
        }
    }
}
int main()
{
//     freopen("a.in","r",stdin);
    scanf( " %d %d ",&n,&m);
     for( int i= 0;i<=n;i++) scanf( " %lf ",&a[i]);
     for( int i= 0;i<=m;i++) scanf( " %lf ",&b[i]);
    m=m+n+ 1;
     int ll= 1;
     for(n= 1;n<m;n*= 2) ll++; ll-= 2; // cout<<ll<<endl;
     for( int i= 0;i<n;i++) rev[i]=(  ( rev[i/ 2]>> 1 )  | ( (i& 1)<<ll) );
    
    FFT(a, 1);
    FFT(b, 1);
     for( int i= 0;i<n;i++)  a[i]=a[i]*b[i];
    FFT(a,- 1);
     for( int i= 0;i<m;i++) printf( " %d  ",( int)(a[i].real()/n+ 0.5));
     return  0;
}