43. Multiply Strings [Leetcode] [FFT] [python]

查看题目
一道大数相乘的题，大数也可以看成是多项式，再运用FFT来处理。

粗略步骤：
1. 将多项式从系数表示转为点值表示
2. 把两个多项式的点值表示相乘
3. 将相乘后的点值表示转为系数表示

---

1. 将多项式从系数表示转为点值表示

系数表示： $A(x) = a_0 + a_1x + a_2x^2 + ··· + a_{n-1}x^{n-1}$
点值表示：$(x_0,A(x_0))、(x_1,A(x_1)、···)$数个这样的点值对

这里我们巧妙地选择点值表示中的x值：
选择1的n次单位根

1的n次单位根：$ω^n = 1$，共有n个不同的根
即$ω^k_n = cos(2πk/n) + i * sin(2πk/n)$
其中$i^2 = -1$

那么$A(ω) = a_0 + a_1ω + a_2ω^2 + ··· + a_{n-1}ω^{n-1}$

系数向量$\vec a = (a_0, a_1, ··· , a_{n-1})$
值向量$\vec A = (A(ω_n^0), A(ω_n^1), ··· , A(ω_n^{n-1}))$

即$\vec A = DFT(\vec a)$
这里我们称为离散傅里叶变换

A1 = dft(a1)
A2 = dft(a2)

2. 把两个多项式的点值表示相乘

for i in range(n):
    A[i] = A1[i]*A2[i]

3. 将相乘后的点值表示转为系数表示

离散傅里叶变换可写为矩阵形式$W * \vec a = \vec A$
即
⎡⎣⎢⎢⎢⎢1111