数论变换NTT
前置知识
- FFT：NTT的思想和FFT一样（[FFT介绍](%3Ca href="https://www.cnblogs.com/guoshaoyang/p/10957854.html"%3Ehttps://www.cnblogs.com/guoshaoyang/p/10957854.html%3C/a%3E)）
概述
数论变换，即NTT（Number Theory Transformation？），是基于数论域的FFT，一般我们默认FFT为负数域上的快速傅里叶变换，和NTT区分。
我们知道，FFT是利用单位复根的周期性，以(Theta(N log N))的复杂度计算(N)组多项式的值。NTT其实就是在数论域上的FFT，它利用素数取模的周期性，达到了和FFT一样的效果。
原根
若(a)模(P)的阶等于(φ(P))，则称(a)为模(P)的一个原根（这个定义不重要）。
对于质数(P)，若(g)是(P)的原根，那么(g^i mod P)的结果两两不同；或者说(g^t = 1 (mod P))当且仅当指数为(t=P-1)的时候成立。
我们发现，原根的性质和单位负数根的性质一一对应，都满足周期性和周期内互异性，故可以用来代入和插值。
NTT
我们有离散傅里叶变换公式
[X_k=sum^{N-1}_{n=0}{x_ne^{-frac{2pi i}{N}nk}} (k=0,1,...,N-1) ]
和逆变换公式
[X_k=frac{1}{N}sum^{N-1}_{n=0}{x_ne^{frac{2pi i}{N}nk}} (k=0,1,...,N-1) ]
我们只需要用原根(g^{frac{P-1}{N}} (mod P))替代单位复根(e^{-frac{-2pi i}{N}})，就可以得到
[X_k=sum^{N-1}_{n=0}{x_ng^{frac{P-1}{N}nk}}(mod P) (k=0,1,...,N-1) ]
和
[X_k=frac{1}{N}sum^{N-1}_{n=0}{x_ng^{-frac{P-1}{N}nk}}(mod P) (k=0,1,...,N-1) ]
实现
和FFT的模板几乎一样，用原根(g^{frac{P-1}{N}} (mod P))替代单位复根(e^{-frac{-2pi i}{N}})即可
#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const int INF=1e9+7,MAXN=3e6+10/*Min:2^20+10*/;
const LL BASE=3,MOD=998244353,INV=332748118;
inline LL fpm(LL base,LL p){
LL ret=1;
while(p){
if(p&1)
ret=ret*base%MOD;
base=base*base%MOD;
p>>=1;
}
return ret%MOD;
}
int N,M,limit=1,lg,rev[MAXN];
inline void NTT(LL *a,int type){
for(int i=0;i<limit;i++)
if(i<rev[i])
swap(a[i],a[rev[i]]);
for(int mid=1;mid<limit;mid<<=1){
int len=mid<<1/*n*/;
LL Wn=fpm(type==1?BASE:INV/*BASE^type*/,(MOD-1)/len);
for(int j=0;j<limit;j+=len){
LL w=1;
for(int k=0;k<mid;k++){
int x=a[j+k],y=w*a[j+k+mid]%MOD;
a[j+k]=(x+y)%MOD;
a[j+k+mid]=(x-y+MOD)%MOD;
w=w*Wn%MOD;
}
}
}
}
LL a[MAXN],b[MAXN];
int main(){
scanf("%d%d",&N,&M);
for(int i=0;i<=N;i++){
scanf("%lld",a+i);
a[i]=(a[i]+MOD)%MOD;
}
for(int i=0;i<=M;i++){
scanf("%lld",b+i);
b[i]=(b[i]+MOD)%MOD;
}
while(limit<=N+M){
limit<<=1;
lg++;
}
for(int i=0;i<limit;i++)
rev[i]=(rev[i>>1]>>1)|((i&1)<<(lg-1));
NTT(a,1);
NTT(b,1);
for(int i=0;i<limit;i++)
a[i]=a[i]*b[i]%MOD;
NTT(a,-1);
LL limit_inv=fpm(limit,MOD-2);
for(int i=0;i<=N+M;i++)
printf("%lld ",a[i]*limit_inv%MOD);
return 0;
}
复杂度
时间
和FFT一样，是(Theta(N log N))，在数据较小的情况下比FFT快，但因为其中有快速幂运算，如果数字太大则会导致TLE
空间
和FFT一样，是(Theta(N))，但注意数组大小不是(N+M)，而是不小于(N+M)的最小的(2)的幂
范围（素数选取）
我们需要一个大素数来取模，这个素数要大于最终输出的答案（但如果太大，就需要使用快速乘）
下面给出常用的几个素数，原根及原根的逆元
| 素数 | 原根 | 逆元 |
|---|---|---|
| 65537 | 3 | 21846 |
| 786433 | 10 | 235930 |
| 5767169 | 3 | 1922390 |
| 7340033 | 3 | 2446678 |
| 23068673 | 3 | 7689558 |
| 104857601 | 3 | 34952534 |
| 167772161 | 3 | 55924054 |
| 469762049 | 3 | 156587350 |
| 998244353 | 3 | 332748118 |
| 1004535809 | 3 | 334845270 |
| 2013265921 | 31 | 64944062 |
| 2281701377 | 3 | 760567126 |
| 3221225473 | 5 | 1932735284 |
| 75161927681 | 3 | 25053975894 |
| 77309411329 | 7 | 22088403237 |
| 206158430209 | 22 | 84337539631 |
| 2061584302081 | 7 | 589024086309 |
| 2748779069441 | 3 | 916259689814 |
| 6597069766657 | 5 | 2638827906663 |
| 39582418599937 | 5 | 15832967439975 |
| 79164837199873 | 5 | 47498902319924 |
| 263882790666241 | 7 | 150790166094995 |
| 1231453023109120 | 3 | 410484341036374 |
| 1337006139375610 | 3 | 445668713125206 |
| 3799912185593850 | 5 | 1519964874237540 |
| 4222124650659840 | 19 | 888868347507335 |
| 7881299347898360 | 6 | 1313549891316390 |
| 31525197391593400 | 3 | 10508399130531100 |
| 180143985094819000 | 6 | 30023997515803300 |
| 1945555039024050000 | 5 | 1167333023414430000 |
| 4179340454199820000 | 3 | 1393113484733270000 |