loj 108 多项式乘法

http://www.elijahqi.win/archives/3253
题目描述

这是一道模板题。

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

输入格式

第一行两个整数 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
模板

ntt

#include<cstdio>
#include<cctype>
#include<algorithm>
#define ll long long
using namespace std;
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;
}
const int N=1e5+10;
const int g=3;
const int mod=998244353;
inline int ksm(ll b,int t){static ll tmp;
    for (tmp=1;t;b=b*b%mod,t>>=1) if (t&1) tmp=tmp*b%mod;return tmp;
}
int a[N<<2],b[N<<2],R[N<<2],n,m;ll invn;
inline int inc(int x,int v){return x+v>=mod?x+v-mod:x+v;}
inline int dec(int x,int v){return x-v<0?x-v+mod:x-v;}
inline void ntt(int *x,int f){
    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){
        ll wn=ksm(g,f==1?(mod-1)/(i<<1):mod-1-(mod-1)/(i<<1));
        for (int j=0;j<n;j+=i<<1){
            ll w=1,t1,t2;
            for (int k=0;k<i;++k,w=w*wn%mod)
                t1=x[j+k],t2=w*x[j+k+i]%mod,x[j+k]=inc(t1,t2),x[j+k+i]=dec(t1,t2);
        }
    }
    if (f==-1) for (int i=0;i<n;++i) x[i]=x[i]*invn%mod;
}
int main(){
    freopen("loj108.in","r",stdin);
    n=read();m=read();
    for (int i=0;i<=n;++i) a[i]=read();
    for (int i=0;i<=m;++i) b[i]=read();
    m+=n;int t=0;for (n=1;n<=m;n<<=1,++t);
    for (int i=0;i<n;++i) R[i]=R[i>>1]>>1|(i&1)<<t-1;
    ntt(a,1);ntt(b,1);invn=ksm(n,mod-2);
    for (int i=0;i<n;++i) a[i]=(ll)a[i]*b[i]%mod;
    ntt(a,-1);
    for (int i=0;i<=m;++i) printf("%d ",a[i]);
    return 0;
}

fft

#include<cmath>
#include<cstdio>
#include<cctype>
#include<algorithm>
#define pi acos(-1)
using namespace std;
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;
}
const int N=1e5+10;
struct C{
    double r,i;
    inline friend C operator+(const C &a,const C &b){return (C){a.r+b.r,a.i+b.i};} 
    inline friend C operator-(const C &a,const C &b){return (C){a.r-b.r,a.i-b.i};}
    inline friend C operator*(const C &a,const C &b){return (C){a.r*b.r-a.i*b.i,a.r*b.i+a.i*b.r};}
    inline void operator*=(const C &a){*this=*this*a;}
}a[N<<2],b[N<<2];
int n,m,R[N<<2]; 
inline void fft(C *x,int f){
    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){
        C wn=(C){cos(pi/i),f*sin(pi/i)};
        for (int j=0;j<n;j+=i<<1){
            C w=(C){1,0},t1,t2;
            for (int k=0;k<i;++k,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;
}
int main(){
    freopen("loj108.in","r",stdin);
    n=read();m=read();
    for (int i=0;i<=n;++i) a[i].r=read();
    for (int i=0;i<=m;++i) b[i].r=read();
    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;
    fft(a,1);fft(b,1);
    for (int i=0;i<n;++i) a[i]*=b[i];fft(a,-1);
    for (int i=0;i<=m;++i) printf("%d ",(int)(a[i].r+0.1));
    return 0;
}