快速傅里叶变换(FFT)

贴过来做个模板。

C为复数类,newArr和at为内存池。

void FFT(int n, C *x)
{
    C W = C(cos(2 * M_PI / n), sin(2 * M_PI / n)), w = C(1), 
        *u = newArr(n >> 1), *v = newArr(n >> 1);
    for (int i = 0; i < n >> 1; ++i)
        u[i] = x[i << 1], v[i] = x[i << 1 | 1];
    if (n > 1) FFT(n >> 1, u), FFT(n >> 1, v);
    for (int i = 0; i < n >> 1; ++i, w *= W)
        x[i] = u[i] + w * v[i], x[i + (n >> 1)] = u[i] - w * v[i];
    at -= n + 5;
}


你可能感兴趣的:(快速傅里叶变换(FFT))