【模板】NTT

\(dalao\)那里扒了个板过来

const int G=3, P=(119<<23)+1;//998244353
int n, m, l, r[Maxn], lim=1, f[Maxn], g[Maxn];
int pow(int x, int k){int res=1; while(k){if(k&1) res=1ll*res*x%P; x=1ll*x*x%P; k >>= 1;} return res;}
void NTT(int *A, int tp){
	for(int i=0; i < lim; i++) if(i < r[i]) swap(A[i], A[r[i]]);
	int gn, g, x, y;
	for(int mid=1, R=mid<<1 ; mid < lim; mid <<= 1){
		gn=pow(G, (P-1)/(mid<<1));
		for(int j=0; j < lim; j+=R){ g=1;
			for(int k=0; k < mid; k++, g=1ll*g*gn%P) 
				x=A[j+k], y=1ll*g*A[j+k+mid]%P, A[j+k]=(x+y)%P, A[j+k+mid]=(x-y+P)%P;
		}
	}
	if(tp == 1) return ;
	int inv=pow(lim, P-2); reverse(A+1, A+lim);
	for(int i=0; i < lim; i++) A[i]=1ll*A[i]*inv%P;
}
void solve(){
	n=read(); m=read();
	for(int i=0; i <= n; i++) f[i]=read(); for(int i=0; i <= m; i++) g[i]=read();
	while(lim <= n+m) lim<<=1, l++; for(int i=0; i < lim; i++) r[i]=r[i>>1]>>1|((i&1)<<l-1);
	NTT(f, 1); NTT(g, 1); for(int i=0; i < lim; i++) f[i]=1ll*f[i]*g[i]%P;
	NTT(f, -1); for(int i=0; i < n+m-1; i++) printf("%d ", f[i]);
}
posted @ 2018-07-14 10:13  zerolt  阅读(113)  评论(0编辑  收藏  举报