[省选前题目整理][UOJ 34]多项式乘法(FFT)

题目链接

http://uoj.ac/problem/34

思路

裸的FFT模板题。

代码

#include <iostream>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <algorithm>
#include <complex>

#define MAXN 1000000
#define PI 3.1415926535897384626

using namespace std;

typedef complex<double> Complex;

inline void rev(Complex a[],int n)
{
    for(int i=1,j=n/2,k;i<n-1;i++)
    {
        if(i<j) swap(a[i],a[j]);
        k=n/2;
        while(j>=k)
        {
            j-=k;
            k>>=1;
        }
        if(j<k) j+=k;
    }
}

inline void FFT(Complex a[],int n,int flag)
{
    rev(a,n);
    for(int i=1;i<n;i<<=1)
    {
        Complex wn=Complex(cos(PI/i),flag*sin(PI/i));
        for(int j=0;j<n;j+=(i<<1))
        {
            Complex w=Complex(1,0);
            for(int k=0;k<i;k++,w=w*wn)
            {
                Complex x=a[j+k],y=w*a[j+k+i]; //!!!!!!
                a[j+k]=x+y;
                a[j+k+i]=x-y;
            }
        }
    }
    if(flag==-1)
        for(int i=0;i<n;i++)
            a[i]=Complex(a[i].real()/n,a[i].imag());
}

Complex a[MAXN],b[MAXN];

int main()
{
    int n,m;
    scanf("%d%d",&n,&m);
    n++,m++;
    for(int i=0;i<n;i++) scanf("%lf",&a[i].real());
    for(int i=0;i<m;i++) scanf("%lf",&b[i].real());
    int N=1,k=0; //2^k=N
    for(;N<=n+m-1;N<<=1,k++);
    FFT(a,N,1);
    FFT(b,N,1);
    for(int i=0;i<N;i++) a[i]*=b[i]; //!!!!!!
    FFT(a,N,-1);
    for(int i=0;i<n+m-1;i++) printf("%d ",(int)(a[i].real()+0.5));
    printf("\n");
    return 0;
}