【模板】多项式乘法（FFT）(FFT板子)

【模板】多项式乘法（FFT）

题意：

输入$n,m(1e6)$
输入$n+1$个数字表示从低到高$F(x)$的系数
输入$m+1$个数表示从低到高$G(x)$的系数
输出：一行$n+m+1$个数表示$F(x)\ast G(x)$从低到高的系数。

代码：

#include<bits/stdc++.h>
#define cp complex<double>
#define pi acos(-1.0)
using namespace std;
const int N=5e6+9;
int n,m,rev[N];
cp f[N],g[N];
int bit;
inline void get_rev(){
    for(int i=0;i<(1<<bit);i++)rev[i]=(rev[i>>1]>>1)|((i&1)<<(bit-1));
}
cp t[N];
void fft(cp *a,int inv){
    int n=1<<bit;
    for(int i=0;i<n;i++)if(i<rev[i])swap(a[i],a[rev[i]]);
    for(int mid=1;mid<n;mid<<=1){
        cp tmp(cos(pi/mid),inv*sin(pi/mid));
        for(int i=0;i<n;i+=mid*2){
            cp base(1,0);
            for(int j=0;j<mid;j++,base*=tmp){
                cp x=a[i+j],y=base*a[i+j+mid];
                a[i+j]=x+y,a[i+j+mid]=x-y;
            }
        }
    }
    if(inv==-1){
        for(int i=0;i<n;i++){cp t(a[i].real()/n,a[i].imag());a[i]=t;}
    }
}
int main(){
  //  freopen("tt.in","r",stdin),freopen("tt.out","w",stdout);
    ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
    cin>>n>>m;
    for(int i=0,x;i<=n;i++){cin>>x;cp t(x,0);f[i]=t;}//不要直接输入cp,会很卡。
    for(int i=0,x;i<=m;i++){cin>>x;cp t(x,0);g[i]=t;}
    while((1<<bit)<(n+m+1))++bit;
    get_rev();
    fft(f,1),fft(g,1);
    for(int i=0;i<(1<<bit);i++)f[i]*=g[i];
    fft(f,-1);
    for(int i=0;i<=m+n;i++)cout<<(int)(f[i].real()+0.5)<<" ";cout<<endl;
    return 0;
}