#108. 多项式乘法

题目描述

这是一道模板题。

输入两个多项式，输出这两个多项式的乘积。

输入格式

第一行两个整数 n n n 和 m m m，分别表示两个多项式的次数。

第二行 n+1 n + 1 n+1 个整数，分别表示第一个多项式的 0 0 0 到 n n n 次项前的系数。

第三行 m+1 m + 1 m+1 个整数，分别表示第二个多项式的 0 0 0 到 m m m 次项前的系数。

输出格式

一行 n+m+1 n + m + 1 n+m+1 个整数，分别表示乘起来后的多项式的 0 0 0 到 n+m n + m n+m 次项前的系数。

样例

样例输入

1 2
1 2
1 2 1

样例输出

1 4 5 2

数据范围与提示

0≤n,m≤105 0 \leq n, m \leq 10 ^ 5 0≤n,m≤10​5​​，保证输入中的系数大于等于 0 0 0 且小于等于 9 9 9。

#include<bits/stdc++.h>
using namespace std;

#define rep(i,a,b) for(int i=a;i<b;i++)
const double pi=acos(-1);
const int maxn=1e5+10;

inline char gc(){
    static char now[1<<16],*S,*T;
    if (T==S){T=(S=now)+fread(now,1,1<<16,stdin);if(T==S) return EOF;}
    return *S++;
}
inline int read(){
    int x=0,f=1;char ch=gc();
    while(!isdigit(ch)) {if(ch=='-') f=-1;ch=gc();}
    while(isdigit(ch)) x=x*10+ch-'0',ch=gc();
    return x*f;
}

struct Com{
    double r,i;
    Com(double _r=0,double _i=0):r(_r),i(_i){}
    Com operator+(Com& b){
        return Com(r+b.r,i+b.i);
    }
    Com operator-(Com& b){
        return Com(r-b.r,i-b.i);
    }
    Com operator*(Com& b){
        return Com(r*b.r-i*b.i,i*b.r+r*b.i);
    }
}a[maxn<<2],b[maxn<<2];
int R[maxn<<2];

int n,m;

//FFT实现
inline void FFT(Com *x,int n,int f){
    //Rader(x,n);
    for (int i=0;i<n;++i) if (i<R[i]) swap(x[i],x[R[i]]);
    for (int i=1;i<n;i<<=1){
        Com wn=Com(cos(pi/i),f*sin(pi/i));
        for (int j=0;j<n;j+=i<<1){
            Com w=Com(1,0),t1,t2;
            for (int k=0;k<i;++k,w=w*wn)
                t1=x[j+k],t2=x[j+k+i]*w,x[j+k]=t1+t2,x[j+k+i]=t1-t2;
        }
    }
    if(f==-1) for (int i=0;i<n;++i) x[i].r/=n;
}

//求卷积
void Conv(Com a[],Com b[],int len){
    FFT(a,len,1);
    FFT(b,len,1);
    for(int i=0; i<len; i++)a[i] = a[i]*b[i];
    FFT(a,len,-1);
}


int main(){
    n=read();m=read();
    rep(i,0,n+1)a[i].r=read();
    rep(i,0,m+1)b[i].r=read();
    
    //补0
    m+=n;int t=0;for (n=1;n<=m;++t,n<<=1);
    for (int i=0;i<n;++i) R[i]=(R[i>>1]>>1)|(i&1)<<t-1;
    //卷积
    Conv(a,b,n);

    rep(i,0,m+1)printf("%d%c",int(a[i].r+0.1),i==m?'\n':' ');
    return 0;
}