FFT入门
前言:
下了好大决心,花了一个晚上的时间,终于看懂了FFT的理论与流程,然后又花了一个晚上实现代码,做了几道模板题。
FFT理论很深,却又很实用,一个很常用的用法就是加速多项式乘法,使得原来O(n^2)的复杂度减小到O(nlogn)。
下面我来大概讲述一下FFT的基本理论与算法流程,帮助初学者了解FFT,同时也是自己的复习。
<1> 介绍:
FFT,全称快速傅里叶变换(fast Fourier transform),是用来计算离散傅里叶变换(DFT)及其逆变换(IDFT)的快速算法。对于DFT,迷一点讲就是说它把时域信号转化为频域信号(不明白也没关系,并不影响学习算法)。在算法竞赛中,我们经常用FFT来加速卷积,做一些多项式乘法或是高精度乘法之类的。
我主要基于多项式乘法来介绍FFT,下面也都是用多项式乘法来讲的~
<2>一些定义:
我们将一个以x为变量的多项式A(x)表示为:
若多项式
若有两个以x为变量的n次多项式
其中
另外,下面还大量提到了复数,并用专用符号
<3>系数表达与点值表达:
对一个次数界为n的多项式
而它的点值表达是一个由n个点值对组成的集合
使得对
对于点值表达,我们可以看成是把多项式
我们可以看到,求出这个多项式的n个点值复杂度为
我们将求值运算的逆称为插值(就是通过点值表达来求出系数表达),有一个定理(差值多项式的唯一性):对于任意n个点值对组成的集合,其中所有的
对于多项式乘法,点值表达是十分方便的。若
若
<4>FFT的大致流程:
现在我们已经能在线性时间内将两个多项式进行乘法操作了,那么剩下的问题就是如何快速实现多项式系数表达与点值表达的快速转化。
上面提到如果巧妙地选取
现在给出FFT算法的大致流程:
*1.加倍次数界:将两个多项式
*2.求值:将系数表达式转化为点值表达式(用2n阶的FFT(DFT),求出多项式在2n次单位复根处的值)。
*3.逐点相乘:将两个多项式点值表达中的
*4.差值:利用积的点值表达再用FFT(IDFT)求出系数表达。
如此,多项式乘法就做完了,我们从两个多项式的系数表达,得到了积的系数表达。
<5>单位复数根:
学FFT,这是极其重要的一点,希望能完全理解其中的变换与结论,最好能手动推算一下,并不难推,大部分是基础的指数运算,但推过与没推过是有很大区别的。
n次单位复根是满足
复数的指数形式的定义:
并且值
称为主n次单位根,所有其它n次单位复根都是
可以得到
消去引理:对任何整数 n>=0,k>=0,以及d>0,
推论:对任意偶数,有
折半引理:如果n>0为偶数,那么n个n次单位复数根的平方的集合就是n/2个n/2次单位复数根的集合。
证明:
根据消去引理,有
求和引理:对任意整数n>=1和不能被n整除的非负整数k,有
注:用等差数列的求和公式推导即可。
<6>DFT:
我们希望计算次数界为n的多项式
我们也记
<7>FFT:
通过FFT,利用单位复数根的特殊性质,我们就可以在
FFT采用分治,将原多项式偶数下标与奇数下标的系数分开,得到两个新的n/2次多项式
其中,
于是有:
于是,就可以递归分治处理了。
下面给出FFT的伪代码:
}
注意:
<8>插值:
现在我们已经知道了如何快速的通过系数表达,在单位复根处快速求出点值,那么,接下来就只剩下最后一项任务,那就是在快速时间内完成DFT的逆过程IDFT,在单位复根处插值,得到结果的系数表达。
我们可以把DFT写成矩阵乘积
定理:
至于证明,只需把矩阵乘法各点的式子写出来,结合求和引理,就可以发现
于是可以推出
其中
我们比较第6节中的式子,发现只要做出以下修改:
*1:把a与y互换
*2:用
*3:将计算结果每个除以n
最后再来一个定理
卷积定理:对任意两个长度为n的向量a和b,其中n是2的幂,
最后,感谢大家的阅读!
如有不足之处,尽可在评论中指出。
本文参考文献:
《算法竞赛入门经典训练指南》——刘汝佳
《算法导论》
(注:本文大部分内容均摘自算法导论,自己加以筛选、注释,第7节中提到的推导过程书中有介绍,若没想出来可以查阅(P534))