FFT算法学习笔记

Posted 2020-10-31 RRRR_wys

写在前边

　　1.辣鸡RRRR_wys之前csdn的博客，千年不更。。。还很水。。。于是开了这个Blog。。。妄图拯救一下自己

　　2.最近接触接触了一些多项式理论。于是翘掉了愉快的高频，通过《算导》稍稍学习了一下

　　3.算法竞赛中，FFT主要解决多项式的乘法等问题

FFT基本概念

　　1.FFT即快速傅里叶变换, 是离散傅里叶变换的加速算法。可以在o(nlogn)的时间内，完成DFT和DFT^-1

　　2.DFT即离散傅里叶变换, 这里主要就是多项式的系数向量转换成点值表示的过程

多项式的表示

　　1.系数表达：对于一个次数上界为n的多项式 A(x)=∑_{(j=0 - n-1)} a_j*x^j 而言，系数表达式便是由系数组成的向量a = (a₀, a₁, a₂, a₃, ... , a_n-1)

         (1)计算A(x₀)的值即：A(x₀) = a₀+x₀(a₁+x₀(a₂+x₀(a₃+x₀(a₄+...+x₀(a_n-2+x₀*a_n-1))...)))  时间为o(n)

　　  (2)通过系数向量计算两个多项式的乘法：设输出向量为c, 而a,b长度n=max(na,nb)那么显然两个c的长度为2n-1，计算方法为：c_j = ∑_{(k=0 ~ j)} a_k*b_j-k，

　　      即向量a,b的卷积

　　2.点值表达式：对于一个次数上界为n的多项式 A(x)=∑_{(j=0 ~ n-1)} a_j*x^j 而言，点值表达式便是由n个点值所构成的集合{ (x₀,y₀), (x₁,y₁), (x₂,y₂), ... , (x_n-1,y_n-1) }

　　    (1)运用之前提到的方法，一个个计算点值时间为o(n²)，之后可以看到如果我们巧妙地选取点x_{k可以使时间变为o(nlogn)}

　　    (2)将点值表达式转换为系数表达式：将已知的点值带入，多项式可已得到一个有n个未知数的n个线性方程

　　　　 a₀ + a₁*x₀ + a₂*x₀² + a₃* x₀³ + ... + a_n-1*x₀^n-1 = y₀ (1)

　　　　 a₀ + a₁*x₁ + a₂*x₁² + a₃* x₁³ + ... + a_n-1*x₁^n-1 = y₁ (2)

　　　　 .......

　　　　 a₀ + a₁*x_n-1 + a₂*x_n-1² + a₃* x_n-1³ + ... + a_n-1*x_n-1^n-1 = y_n-1 (n)

 　　　　 可以写成矩阵形式，既为一个范德蒙矩阵，通过线代知识解这个方程就可以求出系数向量，复杂度o(n³)。另一种方法是采用拉格朗日插值法求解，复杂度o(n²)。因此，n个点的求值运算和插值运算，是定义完备的互逆运算。

　　　 (3)通过点值计算两个多项式的乘积，c = { (x₀,y_A0*y_B0), (x₁,y_A1*y_B1), (x₂,y_A2*y_B2), ... , (x_n-1,y_A(n-1)*y_B(n-1)) }, 可以看到复杂度为o(n)

　　   (4)计算多项式乘法的过程：1）o(nlogn)分别求出两个多项式的点值表达式 2）点值乘法o(n) 3）插值o(nlogn) 4）求出答案多项式的系数表达式

DFT与FFT

　　下面开始讨论，如何在o(nlogn)的时间内，完成求值和插值运算

　　1.单位复数根

    　　n次单位复数根满足ωⁿ=1的复数ω，n此单位复数根的数目恰好有n个，这些根是：ω_n^k = e^2πik/n 这n个根均匀的分布在复平面上，其他n次单位根都是ω_n的幂次。n个n次单位根在乘法意义下形成一个群，该群与模n意义下的加法群，有相同的结构，即ω_nⁱ ω_n^j = ω_n^{(i + j) mod n}

　　　(1) 消去引理：ω_nd^kd =ω_n^k  (n>=0, k>=0, d>0)

　　　(2) ω_n^n/2 =ω₂ =-1

　　　(3) 折半引理：如果n>0为偶数，那么n个n次单位复数根的平方的集合就是n/2个n/2次单位复数根的集合

　　　(4) 求和引理：∑_{(j=0 ~ n-1)} (ω_n^k)^j = 0 (n>=1, 不能被n整除的非负整数k)

　　2.DFT

　　我们希望计算ω_n⁰, ω_n¹, ω_n², .... , ω_n^n-1处的值（在多项式乘法中其实是2n个点值）。假设给定系数a=(a₀, a₁, a₂, ... , a_n-1), 即对于k=0, 1, 2, 3, ... , n-1求出y_k = A(ω_n^k)，记向量y=(y₁, y₂, y₃, ... ,y_n-1)就是系数向量a的离散傅里叶变换DFT，记为 y = DFT_n(a)

　　3. FFT

　　通过快速傅里叶变换FFT，利用单位复数根的性质，我们可以在o(nlogn)的时间内计算DFT_n(a)。首先假设次数n恰好是2的整数幂，不足在高次位置项添0。

　　定义两个新的次数界为n/2的多项式 A_[0] = a₀ + a₂*x + a₄*x² + a₆* x³ + ... + a_n-2*x^n/2-1 , A_[1] = a₁ + a₃*x + a₅*x² + a₇* x³ + ... + a_n-1*x^n/2-1

　　A_[0]包含所有偶数下标的系数，A_[1]包含所有奇数下标的系数，于是有A(x) = A_[0](x2) + x*A_[1](x²) 。

　　所以，求解A(x)的DFN，转换为了求解多项式A_[0], A_[1]在(ω_n⁰)², (ω_n¹)², (ω_n²)² , ... , (ω_n^n-1)² 处的取值。

　　根据折半引理，(ω_n⁰)², (ω_n¹)², (ω_n²)² , ... , (ω_n^n-1)² 并不是由n个不同的值组成，而是由n/2个n/2次单位根复数根所组成，每个根正好出现两次。因此，我们递归的对次数界为n/2的多项式A_[0], A_[1]在n/2个n/2次单位复数根处进行求值。