• 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$


  • 相关阅读:
    CodeForces 1017B
    POJ 1323-Game Prediction(贪心)
    CodeForces 104B-Testing Pants for Sadness(思维题)
    CodeForces 1324D
    UVA 1152-4 Values whose Sum is 0(二分查找)
    POJ 1700-Crossing River(贪心)
    HDU 1789-Doing Homework again(并查集+贪心)
    SDUT 1298-活动选择(贪心)
    HDU 1272-小希的迷宫(带环并查集)
    CodeForces
  • 原文地址:https://www.cnblogs.com/jianglangcaijin/p/6000204.html
Copyright © 2020-2023  润新知