NTT快速数论变换

NTT

$NTT$和$FTT$十分相似，在$FTT$中使用了单位复根$\omega$来取函数值，在$NTT$中则使用原根$g$来取函数值。在$NTT$中，所有的数都是模意义下的。

$g$​的一些性质

1. $g^1,g^2,\cdots ,g^{p-1}$在模$p$意义下互不相同
2. 根据欧拉定理可知$g^{p-1}\equiv 1$

用$g$来替换$\omega$实现$FFT$到$NTT$的转化

其实只要$g$能够在模意义下实现$\omega$的性质就行了。

假设$p=qn+1$​​，其中$n$​​是一个$2$​​的幂，我们钦定在对$f_n(x)$进行$NTT$​​时使用的单位原根是$g_n$，其计算方式为$g_n=g^q=g^{\frac{p-1}{n}}$

下面考虑$\omega$在$FFT$中使用到的性质：

1. ${\omega }_{2n}^{2i}={\omega }_n^i$，$g_{2n}^{2i}=g^{\frac{p-1}{2n}\cdot 2i}=g^{\frac{p-1}{n}\cdot i}=g_n^i$
2. ${\omega }_n^{\frac{n}{2}}=-1$​，$g_n^{\frac{n}{2}}=g^{\frac{p-1}{2}}=\sqrt{g^{p-1}}=\sqrt{1}$，根据二次探索定理可知，在模意义下$\sqrt{1}$只有两种取值，即$1$或$p-1$，根据原根的性质一排除取值为$1$的可能，那么$g_n^{\frac{n}{2}}=\sqrt{1}=p-1=-1$
3. $\forall k\in Z,{\omega }_n^{kn}=1$，$\forall k\in Z,g_n^{kn}=(g_n^n{)}^k=(g^{p-1}{)}^k=1$​

从$FFT$到$NTT$

我们只要把单位复根替换为单位原根即可，一切计算都在模意义下进行

#include<bits/stdc++.h>
using namespace std;
const int mod=998244353;
const int N=1<<22;
int f[N],g[N],h[N];
int r[N];
int nf,ng,nh=1;
int ksm(int x,int y){
	int res=1;
	while (y){
		if (y&1) res=1ll*res*x%mod;
		x=1ll*x*x%mod;
		y>>=1;
	}
	return res;
}
void FFT(int *f,int rev){//rev=1->DFT  rev=-1->IDFT
	for (int i=0;i<nh;++i) if (r[i]<i) swap(f[i],f[r[i]]);
	for (int l=2;l<=nh;l<<=1){
		for (int i=0;i<nh;i+=l){
			int step=ksm(3,(mod-1)/l),mul=1;
			for (int j=0;j<(l>>1);++j){
				int tmp=f[i+j+(l>>1)];
				f[i+j+(l>>1)]=(1ll*f[i+j]-1ll*mul*tmp%mod+mod)%mod;
				f[i+j]=(1ll*f[i+j]+1ll*mul*tmp%mod)%mod;
				mul=1ll*mul*step%mod;
			}
		}
	}
	if (rev==-1){
		reverse(f+1,f+nh);
		int inv=ksm(nh,mod-2);
		for (int i=0;i<nh;++i) f[i]=1ll*f[i]*inv%mod;
	}
}
int main(){
	scanf("%d%d",&nf,&ng);
	while (nh-1<nf+ng) nh<<=1;
	for (int i=0;i<=nf;++i) scanf("%d",&f[i]);
	for (int i=0;i<=ng;++i) scanf("%d",&g[i]) ;
	r[0]=0; 
	for (int i=1;i<nh;++i) r[i]=(r[i>>1]>>1)|(i&1?(nh>>1):0);
	FFT(f,1);
	FFT(g,1);
	for (int i=0;i<nh;++i) h[i]=1ll*f[i]*g[i]%mod;
	FFT(h,-1);
	for (int i=0;i<=nf+ng;++i) printf("%lld ",h[i]);
}