C++怎么实现一个快速傅里叶变换(FFT)_C++信号处理与数值计算算法

快速傅里叶变换(FFT)基于分治思想,采用迭代与位逆序置换实现高效DFT计算。1. 使用std::complex表示复数,利用单位根ω_N^k的周期性加速运算;2. 通过位逆序置换预处理输入,如8点FFT下标重排为[0,4,2,6,1,5,3,7],确保内存连续访问;3. 迭代实现中,从长度2开始逐层合并,每层用单位根旋转因子更新值,支持原地计算;4. 应用于多项式乘法时,将系数转为频域相乘再逆变换,时间复杂度O(n log n)。

c++怎么实现一个快速傅里叶变换(fft)_c++信号处理与数值计算算法

快速傅里叶变换(FFT)是离散傅里叶变换(DFT)的高效算法,广泛用于信号处理、多项式乘法和数值计算。C++中实现FFT通常采用分治思想,最常见的是基于“**迭代+位逆序置换**”的Cooley-Tukey算法。

1. 基本原理与复数支持

FFT处理的是复数序列,因此需要使用std::complex来表示复数。核心是将长度为N(要求N是2的幂)的DFT分解为更小的DFT,利用单位根的周期性和对称性减少计算量。

单位根定义为:ω_N^k = exp(-2πi * k / N),其中i是虚数单位。

2. 位逆序置换(Bit-reversal Permutation)

递归版FFT自然完成子问题划分,但迭代版需要预先将输入数组按位逆序重排。例如,8点FFT中,下标二进制表示如下:

立即学习“C++免费学习笔记(深入)”;

0: 000 → 000: 01: 001 → 100: 42: 010 → 010: 23: 011 → 110: 64: 100 → 001: 15: 101 → 101: 56: 110 → 011: 37: 111 → 111: 7

重排后顺序为 [0,4,2,6,1,5,3,7],这样每一层合并都能连续访问内存。

3. 迭代实现FFT

以下是一个完整的C++实现,支持原地计算:

#include #include #include #include 

using namespace std;using Complex = complexconst double PI = acos(-1);

// 位逆序置换void bitReverse(vector& a) {int n = a.size();int bits = 0;while ((1 << bits) < n) bits++;

for (int i = 0; i < n; i++) {    int rev = 0;    for (int j = 0; j < bits; j++)        if (i & (1 << j))            rev |= 1 << (bits - 1 - j);    if (i < rev)        swap(a[i], a[rev]);}

}

// 快速傅里叶变换(原地迭代版)void fft(vector& a, bool invert) {int n = a.size();bitReverse(a);

for (int len = 2; len <= n; len <<= 1) {    double angle = 2 * PI / len * (invert ? 1 : -1);    Complex wlen(cos(angle), sin(angle));    for (int i = 0; i < n; i += len) {        Complex w(1);        for (int j = 0; j < len / 2; j++) {            Complex u = a[i + j];            Complex v = a[i + j + len/2] * w;            a[i + j] = u + v;            a[i + j + len/2] = u - v;            w *= wlen;        }    }}if (invert) {    for (Complex& x : a)        x /= n;}

}

4. 使用示例:多项式乘法

FFT常用于高效计算两个多项式的卷积(即系数乘法):

vector multiply(const vector& a, const vector& b) {    vector fa(a.begin(), a.end()), fb(b.begin(), b.end());    int n = 1;    while (n < a.size() + b.size())        n <<= 1;
fa.resize(n); fb.resize(n);fft(fa, false);fft(fb, false);for (int i = 0; i < n; i++)    fa[i] *= fb[i];fft(fa, true);vector result(n);for (int i = 0; i < n; i++)    result[i] = round(fa[i].real()); // 取实部并四舍五入return result;

}

输入两个系数向量,输出它们的卷积结果,时间复杂度从O(n²)降至O(n log n)。

基本上就这些。注意FFT要求长度为2的幂,如果不是可补零扩展。该实现稳定且易于集成到信号处理流程中。

以上就是C++怎么实现一个快速傅里叶变换(FFT)_C++信号处理与数值计算算法的详细内容,更多请关注创想鸟其它相关文章!

版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 chuangxiangniao@163.com 举报,一经查实,本站将立刻删除。
发布者:程序猿,转转请注明出处:https://www.chuangxiangniao.com/p/1485976.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2025年12月19日 09:24:03
下一篇 2025年12月19日 09:24:09

相关推荐

发表回复

登录后才能评论
关注微信