[多项式全家桶]FFT/NTT

开始补EC PK赛的题目，在此之前决定先完善多项式全家桶的学习任务。

---

FFT

在$O(logN)$的时间完成点值、系数的多项式表示，在$O(N)$的时间完成多项式乘积。

FFT在之前学习过：[多项式全家桶] 快速傅立叶变换(FFT)
再推荐一篇博客：FFT\NTT总结

在学习别的知识点之前又自己结合原理撸了一遍fft代码，贴一下符合自己习惯(比如从0开始的计数)的代码：

//逆fft取op==-1是因将范德蒙德行列式的逆时每一项取共轭再除以n
void fft(COM *a, int op){
    for(int i = 0; i < N; i++) if(i < inv[i]) swap(a[inv[i]], a[i]);
    
    for(int i = wei - 1; i >= 0; i--){
	    //d表示第i层每一小段的长度
        int d = 1 << (wei - i);				
        //wn是当前层下使用的单位根
        COM wn = {cos(2 * pi / d), op * sin(2 * pi / d)};
        
        //j遍历第i层的每段开头，更新每个d小段
        for(int j = 0; j < N; j += d){
			//在一个小段中新设一个量w进行更新操作
            COM w = {1, 0};
            for(int k = 0; k < d / 2; k++){
                COM x, y; 
                x = a[j + k]; y = a[j + k + d / 2] * w;
                a[j + k] = x + y; a[j + k + d / 2] = x - y;
                w = w * wn;
            }
        }
    }
}

注意，如果对多项式A和B进行FFT求乘积时，A、B数组的大小都要开至maxn << 2!!

---

NTT

在此之前，需要学习一下原根的知识点。

参考博客：HolseLee

定义阶为：设$m>1,gcd(a,m)=1,$那么使得$a^r\equiv 1(mod$ $m)$成立的最小正整数称为$a$对模$m$的阶，记为${\delta }_m(a)$

易知阶有两个相关定理：

• 若满足$m>1,gcd(a,m)=1$，并且$a^n\equiv 1(mod$ $m)$，则有${\delta }_m(a)$ | $n.$
• ${\delta }_m(a)$ | $\varphi (m).$

进而，定义原根：设$a$是整数，$m$是正整数，如果${\delta }_m(a)=\varphi (m)$，那么称$a$为$m$的一个原根。

原根有三个相关定理：

• 一个正整数$m$有原根的充要条件是$m=2,4,p^e,2p^e$，其中$p$是奇素数，$e$是正整数。
  证明略
• 若$p$是$m$的一个原根，则$$p,p^2,p^2,\dots ,p^{\varphi (m)}$$各数对$m$取模的最小剩余就是小于$m$且与$m$互素的$\varphi (m)$个数的一个排列。
  证明: 易知任意两个元素均不相同，因而接下来只需要证明每个元素都与$m$互素。
  由$p^{\varphi (m）}\equiv 1(modm)$可知$$(p^i{)}^{\varphi (m)}\equiv 1(modm)$$因而我们只需要证明如果某个数$a$对$m$的阶存在，那么就有$gcd(a,m)=1$.为此，我们假设$a^i\equiv 1(modm)$且由$gcd(a,m)=d$.所以有$$a^i=km+1$$
  其中$a^i$和$km$均可以被$d$整除，因而1也必须被d整除，故$d\mid 1$即$d=1.$
• 对每一个正整数$m$都有$\varphi (\varphi (m))$个原根，特别地，素数$p$有$\varphi (p-1)$个原根。
  证明: 若$p$是$m$的一个原根，那么由上述定理其既约剩余系取遍小于$m$且与$m$互素的所有数。在$1、2、\dots 、\varphi (m)$中只有那些与$\varphi (m)$互素的幂次才可以成为原根，这是因为那些与$\varphi (m)$不互素的元素产生的$\varphi (m)$个元素是某些元素集合的$d$次重复，其中$d=gcd(p^i,\varphi (m)).$

原根的求法？
首先将$\varphi (m)$做标准素数分解:$$\varphi (m)=p_1^{e_1}\ast p_2^{e_2}\ast \cdots \ast p_k^{e_k}$$
然后枚举$g$，若恒满足$$g^{\frac{\varphi (m)}{p_i}}\ne 1(modm),i=1,2,\dots ,k$$
则$g$是$m$的一个原根。

证明可以从阶的第二个定理出发。因为$\delta m(a)$总是$\varphi (m)$的一个因子，所以若$\delta m(a)$小于$\varphi (m)$，则$\delta m(a)$一定会小等于某一个$\frac{\varphi (m)}{p_i}$，从而$g^{\frac{\varphi (m)}{p_i}}$模$m$的值一定为1.

下面正式学习一下NTT(快速数论变换)

参考博客：快速数论变换(NTT)小结

类似于FFT中我们寻找的$w_n$，我们在模$M$下利用原根做循环。

因为我们采取二分递归的办法解决点值、系数转化问题，所以我们希望找一些具有优良的特殊性质的素数$M=k\ast 2^r+1$作为模数。(参考：FFT用到的各种素数)
比如：$119<<23+1$可以处理$1e6$的数据,$17<<27+1$平方后不会爆longlong. 这两个数字对应的原根均为3.

之后，我们构造循环：$$w_n=p^{\frac{m-1}{n}}$$
由原根性质，$w_n,w_n^2,\dots ,w_n^n$构成一个$n$阶循环群，满足复数$w_n=e^{\frac{2\pi }{n}}$的对应性质。
所以在求做INNT的时候用$p$的逆作为逆矩阵的元素，最后在输出答案的时候除以$n$，即乘上$n$的逆。

核心代码与FFT类似，仅需要在必要处取模&更改对应的循环因子$w_n$即可。

---

暂且先复习FFT和学习NTT，之后更多的多项式算法日后再学。

ywwyww
DCDCBigBig