NTT

1 问题描述
FFT问题解决的是复数域上的卷积。如果现在的问题是这样:给出两个整数数列$Ai,Bj,0leq ileq n-1,0leq jleq m-1$,以及素数$P$,计算新数列$Ci=(sum_{k}A_{i-k}B_{k})\%P$。不在$A,B$定义域内的值均为0.NTT就是解决这样在模意义下的卷积问题。

2 预备知识
原根的概念:对于两个正整数$a,m$,如果$Gcd(a,m)=1$,那么存在$dleq m-1$(比如$varphi (m)$)使得$a^{d}\%m=1$。如果对于某个g使得在$[1,m-1]$内只有$varphi (m)$满足$g^{varphi (m)}\%m=1$,那么g被称作m的原根。比如7的原根是3,因为在1,2,3,4,5,6中只有3^6%7=1。

3 算法描述
NTT的原理与FFT的原理完全相同。即首先我们会将$A,B$通过NTT变换转化成新数列$A^{'},B^{'}$,然后我们有$C_{k}^{'}=A_{k}^{'}B_{k}^{'}$。最后通过NTT逆变换,将$C^{'}$转化成$C$.

FFT变换是这样的:$A_{k}^{'}=sum_{j=0}^{n-1}A_{j}e^{frac{2pi i}{n}kj}=sum_{j=0}^{n-1}A_{j}w_{n}^{kj}$。

FFT逆变换是这样的:$A_{k}=frac{1}{n}sum_{j=0}^{n-1}A_{j}^{'}e^{-frac{2pi i}{n}kj}=frac{1}{n}sum_{j=0}^{n-1}A_{j}^{'}w_{n}^{-kj}$

$e^{frac{2pi i}{n}}$在这里之所以能够使用,是因为它在乘法意义下构成一个群。同理,在模P的系统中,对于乘法来说,$g^{frac{P-1}{n}}$也构成一个群,类似$(e^{frac{2pi i}{n}})^{n}=1$,$(g^{frac{P-1}{n}})^{n}%P=1$。其中g是P的原根。

因此,NTT变换是: $A_{k}^{'}=sum_{j=0}^{n-1}A_{j}g^{frac{P-1}{n}kj}$

NTT逆变换: $A_{k}=frac{1}{n}sum_{j=0}^{n-1}A_{j}^{'}g^{-frac{P-1}{n}kj}=frac{1}{n}sum_{j=0}^{n-1}A_{j}^{'}g^{((P-1)-frac{P-1}{n})kj}$

所以,NTT变换使用的基为$g^{frac{P-1}{n}}$,NTT逆变换使用的基为$g^{(P-1)-frac{P-1}{n}}$

这里有一些需要注意的问题。在实际计算时,我们会将整个数列的长度扩展至$2^{K}$次幂,因此在每一次分治时,每个子问题的长度n也都是2的某次幂。而这里,必须要保证n能够整除$P-1$,所以若$P=c*2^t+1$,其中$c$为奇数。那么一定要满足$tgeq K$


原文地址:https://www.cnblogs.com/jianglangcaijin/p/6000204.html