声明:参考了 y y c yyc yyc 的 blog 和 PPT (from smwc) ,以及 w z r wzr wzr 的 blog 。
目录
- Part 1 多项式
- Part 2 FFT概论
- Part 3 点值与插值
- Part 4 复数,单位根
- Part 5
Part 1 多项式
- 定义:对于有限数列 A 0 A_{0} A0~ n _{n} n ,形如 A ( x ) = a 0 + a 1 x + a 2 x 2 + . . . + a n x n A(x)=a_0+a_1x+a_2x^2+ ... +a_nx^n A(x)=a0+a1x+a2x2+...+anxn (即 A ( x ) = ∑ i = 0 n A i x i A(x)= \sum_{i=0}^{n} A_ix^i A(x)=∑i=0nAixi) 的函数称为多项式。记 A ( x ) A(x) A(x) 的 x n x^n xn 项系数为 [ x n ] A ( x ) [x^n]A(x) [xn]A(x) ,简写为 A [ n ] A[n] A[n] 。
- 运算:
- 加法: C ( x ) = A ( x ) + B ( x ) = ∑ i = 0 n ( A i + B i ) x i C(x)=A(x)+B(x) = \sum_{i=0}^{n}(A_i+B_i)x^i C(x)=A(x)+B(x)=∑i=0n(Ai+Bi)xi
- 减法: C ( x ) = A ( x ) − B ( x ) = ∑ i = 0 n ( A i − B i ) x i C(x)=A(x)-B(x) = \sum_{i=0}^{n}(A_i-B_i)x^i C(x)=A(x)−B(x)=∑i=0n(Ai−Bi)xi
- 乘法: C ( x ) = A ( x ) × B ( x ) = ∑ i = 0 n ∑ j = 0 m A i B j x i + j C(x)=A(x) \times B(x) = \sum_{i=0}^{n} \sum_{j=0}^{m} A_iB_jx^{i+j} C(x)=A(x)×B(x)=∑i=0n∑j=0mAiBjxi+j
- 卷积:某二元运算 ⊕ \oplus ⊕ 的定义域和值域均为 N \mathbb{N} N ,数列 A , B A,B A,B 的 ⊕ \oplus ⊕卷积 为一个新数列 C C C ,有 C [ k ] = ∑ i ⊕ j = k A [ i ] B [ j ] C[k]= \sum_{i \oplus j=k} A[i]B[j] C[k]=∑i⊕j=kA[i]B[j] ( ⊕ \oplus ⊕ 可以代表任意运算符)。容易发现,多项式乘法就是“加法卷积”。
Part 2 FFT概论
对于两个多项式的乘法,有简单粗暴的 O ( n 2 ) O(n^2) O(n2) 级别方法。但这只适用于少部分题目,于是某个天才就发明的一种新方法,叫 Fast Fourier Transformation(FFT)。利用卷积思想,化乘为加,快速计算多项式乘法,只需 O ( n l o g n ) O(nlogn) O(nlogn) 。
Part 3 点值与插值
- 点值:将多项式 A ( x ) A(x) A(x) 视为函数,对于常数 t 0 t_0 t0 , A ( t 0 ) = ∑ i = 0 + ∞ A [ i ] t 0 i A(t_0)=\sum_{i=0}^{+\infty} A[i]t_0^i A(t0)=∑i=0+∞A[i]t0i 为 A ( x ) A(x) A(x) 在 t 0 t_0 t0 处的点值。
- 作用:已知 A ( x ) , B ( x ) , C ( x ) A(x),B(x),C(x) A(x),B(x),C(x),可以快速判断 C ( x ) = A ( x ) B ( x ) C(x)=A(x)B(x) C(x)=A(x)B(x) ;
- 核心:通过选择具体的常数,将复杂的多项式乘法转换成数的乘法,提升效率。
- 插值:已知 A ( x ) A(x) A(x) 的若干点,求 A ( x ) A(x) A(x) 的系数序列。
将上面两种表示法组合起来,就有一个 “求值-插值法” ,得到多项式乘法一个新想法。即,把系数表达转换为点值表达 (DFT) -> 点值表达相乘 -> 把点值表达转换为系数表达 (IDFT)。
这就是 F F T FFT FFT 的思路:
- 选择一个足够大的 n n n,选定 n n n 个不同的数 x 1 , x 2 , x 3 , . . . , x n x_1,x_2,x_3, ... ,x_n x1,x2,x3,...,xn ;
- 求出点值 A ( x 1 ) , A ( x 2 ) , . . . , A ( x n ) A(x_1),A(x_2),... ,A(x_n) A(x1),A(x2),...,A(xn) 和 B ( x 1 ) , B ( x 2 ) , . . . , B ( x n ) B_(x_1),B(x_2),...,B(x_n) B(x1),B(x2),...,B(xn) ;
- 根据 C ( x i ) = A ( x i ) B ( x i ) C(x_i)=A(x_i)B(x_i) C(xi)=A(xi)B(xi) 得到 C ( x 1 ) , C ( x 2 ) , . . . , C ( x n ) C(x_1),C(x_2),...,C(x_n) C(x1),C(x2),...,C(xn) :
- 插值求出 C ( x ) C(x) C(x) 的系数;
Part 4 复数,单位根
考虑如何做 FFT 的第一步(即选择 n n n 个不同的数),一般直觉会选正整数等一些熟悉的数,但这没什么性质(加速不了多项式乘法)。于是单位根就登场了……
- 复数:令虚单位 i i i 满足 i 2 = − 1 i^2=-1 i2=−1,形如 a + b i a+bi a+bi ( a , b ∈ R a,b \in \mathbb{R} a,b∈R) 的数称为复数。
将复数 z = a + b i z=a+bi z=a+bi 看作平面直角坐标系中的点 P ( a , b ) P(a,b) P(a,b) ,原点 O ( 0 , 0 ) O(0,0) O(0,0),称表示复数的平面直角坐标系为复平面, r = ∣ O P ∣ r=|OP| r=∣OP∣ 为 z z z 的模长,射线 O P OP OP 与 x x x轴正半轴的夹角 θ \theta θ 为 z z z 的辐角。有 z = r ( c o s θ + i z=r(cos \theta + i z=r(cosθ+i s i n θ ) sin \theta) sinθ) 。- 复数加法: ( a + b i ) + ( c + d i ) = ( a + c ) + ( b + d ) i (a+bi)+(c+di) = (a+c) + (b+d)i (a+bi)+(c+di)=(a+c)+(b+d)i
- 复数减法: ( a + b i ) − ( c + d i ) = ( a − c ) + ( b − d ) i (a+bi)-(c+di) = (a-c) + (b-d)i (a+bi)−(c+di)=(a−c)+(b−d)i
- 复数乘法: ( a + b i ) × ( c + d i ) = a c − b d + ( a d + b c ) i (a+bi) \times (c+di) = ac-bd+(ad+bc)i (a+bi)×(c+di)=ac−bd+(ad+bc)i
- 复数除法: ( a + b i ) (a+bi) (a+bi) / / / ( c + d i ) = a c + b d c 2 + d 2 + b c − a d c 2 + d 2 i (c+di)= \frac{ac+bd}{c^2+d^2} + \frac{bc-ad}{c^2+d^2}i (c+di)=c2+d2ac+bd+c2+d2bc−adi
- 复数的共轭: a + b i a+bi a+bi 的共轭为 a − b i a-bi a−bi
- 单位根: n n n 次本原单位根记为 ω n \omega ^n ωn ( n ∈ N n \in \mathbb{N} n∈N),单位根为复数,满足 ω n = 1 \omega ^n=1 ωn=1 ,且 ω 0 , ω 1 , ω 2 , . . . , ω n − 1 \omega ^0,\omega ^1,\omega ^2 ,...,\omega ^{n-1} ω0,ω1,ω2,...,ωn−1 互不相同。
引入 单位圆(圆心在原点且半径为 1 1 1 的圆):
记第 n n n 个 n n n 次单位根为 ω n n − 1 。 \omega _n^{n-1}。 ωnn−1。有 ω n k = k ( c o s 2 π n + i \omega _n^k=k(cos\frac{2\pi}{n}+i ωnk=k(cosn2π+i s i n 2 π n ) sin \frac{2\pi}{n}) sinn2π) 。
ω n \omega _n ωn 是模长为 1 1 1 ,圆上的点所表示的复数模长都为 1 1 1 ,所以单位根一定在单位圆上。又有 ω n \omega _n ωn 的辐角为 2 π n \frac{2\pi}{n} n2π (即 1 n \frac{1}{n} n1 圆周),乘一次 ω n \omega _n ωn 相当于绕原点逆时针旋转 1 n \frac{1}{n} n1 圆周,乘 n n n 次恰好转完一圈回到原地,满足 “ ω n = 1 \omega ^n=1 ωn=1 ,且 ω 0 , ω 1 , ω 2 , . . . , ω n − 1 \omega ^0,\omega ^1,\omega ^2 ,...,\omega ^{n-1} ω0,ω1,ω2,...,ωn−1 互不相同” 这一条件。
- 单位根的一些性质:
- ω n 0 = 1 \omega _n^0 =1 ωn0=1
- ω n k = ( ω n 1 ) k \omega _n^k = (\omega _n^1)^k ωnk=(ωn1)k
- ω n i × ω n j = ( ω n 1 ) i × ( ω n 1 ) j = ω n i + j \omega _n^i \times \omega_n^j =(\omega_n^1)^i \times (\omega_n^1)^j = \omega_n^{i+j} ωni×ωnj=(ωn1)i×(ωn1)j=ωni+j
- ω n − k = ω n n − k \omega_n^{-k} = \omega_n^{n-k} ωn−k=ωnn−k
- ω 2 n 2 k = ω n k = ω p n p k \omega_{2n}^{2k} = \omega_n^k = \omega_{pn}^{pk} ω2n2k=ωnk=ωpnpk
- ω n ( k + n / 2 ) = − ω n k \omega_{n}^{(k+n/2)} = - \omega_{n}^{k} ωn(k+n/2)=−ωnk