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

来源：互联网发布：java spring mvc面试题编辑：程序博客网时间：2024/06/18 10:12

题目链接

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.1415926535897384626using 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;}

0 0