[学习笔记]快速傅里叶变换

前言:今天遇到各种烦心事,包括但不限于自己的企鹅号登不上去了,一堆高中摆烂人借着会考训练的名义在我训练的机房打游戏。

我直接跳过大部分步骤吧。

做一个类似于总结类的博客得了。

首先\(DFT\)是将一个多项式转为一个特定的点值表达的形式。

我们在复数域选取单位根作为点值的\(x\)。

有以下三个重要结论:
\(w_n^{n} = 1\)
\(w_n^{k} = w_{2n}^{2k}\)
\(w_{2n}^{k + n} = -w_{2n}^{k}\)

考虑我们在处理\(F(x)\)的\(DFT\)时

我们将其偶数和奇数位的系数单独提出来作为一个多项式\(G,H\)。

那么有\(F(x) = G(x^2) + x \times H(x ^ 2)\)。

那么对于\(DFT\)来说

\(DFT(F)_k(F(w_{n}^k)) = G((w_{n}^{k})^2) + w_n^{k} \times H((w_{n}^{k})^2)\\=G((w_{n}^{2k})) + w_n^{k} \times H((w_{n}^{2k}))\\=G((w_{n/2}^{k})) + w_n^{k} \times H((w_{n/2}^{k}))\\=DFT(G)_k + w_n^{k}\times DFT(F)_k\)

同理有

\(DFT(F)_{k + n / 2}(F(w_{n}^{k + n / 2})) = DFT(G)_k - w_n^{k}\times DFT(F)_k\)

递归FFT
#include <cmath>
#include <complex>

typedef std::complex<double> Comp;  // STL complex

const Comp I(0, 1);  // i
const int MAX_N = 1 << 20;

Comp tmp[MAX_N];

void DFT(Comp *f, int n, int rev) {  // rev=1,DFT; rev=-1,IDFT
  if (n == 1) return;
  for (int i = 0; i < n; ++i) tmp[i] = f[i];
  for (int i = 0; i < n; ++i) {  // 偶数放左边,奇数放右边
    if (i & 1)
      f[n / 2 + i / 2] = tmp[i];
    else
      f[i / 2] = tmp[i];
  }
  Comp *g = f, *h = f + n / 2;
  DFT(g, n / 2, rev), DFT(h, n / 2, rev);  // 递归 DFT
  Comp cur(1, 0), step(cos(2 * M_PI / n), sin(2 * M_PI * rev / n));
  // Comp step=exp(I*(2*M_PI/n*rev)); // 两个 step 定义是等价的
  for (int k = 0; k < n / 2; ++k) {
    tmp[k] = g[k] + cur * h[k];
    tmp[k + n / 2] = g[k] - cur * h[k];
    cur *= step;
  }
  for (int i = 0; i < n; ++i) f[i] = tmp[i];
}

蝴蝶变换,等我被卡了我再来填这个坑。

上一篇:计算机系统基础复习笔记——数据的机器级表示与处理


下一篇:CF1036A Function Height 题解