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;
}