C++实现快速傅里叶变换FFT（附带源码）

最新推荐文章于 2025-04-30 15:36:57 发布

Katie。

最新推荐文章于 2025-04-30 15:36:57 发布

阅读量1k

点赞数 21

分类专栏： c、c++实战项目文章标签： c++ 开发语言

本文链接：https://blog.csdn.net/m0_61840987/article/details/145400021

版权

c、c++实战项目专栏收录该内容

516 篇文章

订阅专栏

C++ 实现快速傅里叶变换（FFT）

项目介绍

快速傅里叶变换（Fast Fourier Transform, FFT）是一种高效计算 离散傅里叶变换（DFT） 及其逆变换的方法。傅里叶变换在信号处理、图像处理、数据分析等领域具有广泛应用，例如：

音频频谱分析
图像处理中的滤波
无线通信中的调制解调
数据压缩（如 JPEG、MP3）

本项目将使用 C++ 实现 基于 Cooley-Tukey 算法的 FFT，并提供：

FFT 计算（递归和迭代方式）
逆 FFT（IFFT）计算
FFT 在信号处理中的应用
完整代码、详细注释及代码解读

项目实现思路

离散傅里叶变换（DFT）公式：

DFT 计算复杂度为 O(N²)，FFT 通过 分治思想 将其优化至 O(N logN)，主要思路如下：

奇偶分解：将原始序列按奇数索引和偶数索引拆分。
递归计算：分别对奇偶部分求 FFT。
合并结果：利用旋转因子合并结果。

本项目提供 递归版和迭代版 FFT，并支持 逆变换（IFFT）。

代码实现

#include <iostream>
#include <vector>
#include <complex>
#include <cmath>

using namespace std;

// 定义复数类型
using Complex = complex<double>;
using CArray = vector<Complex>;

/**
 * @brief 递归实现快速傅里叶变换 (FFT)
 * @param x 输入复数数组
 */
void fft(CArray &x) {
    int N = x.size();
    if (N <= 1) return; // 递归终止条件

    // 拆分为偶数和奇数部分
    CArray even(N / 2), odd(N / 2);
    for (int i = 0; i < N / 2; i++) {
        even[i] = x[i * 2];
        odd[i] = x[i * 2 + 1];
    }

    // 递归计算 FFT
    fft(even);
    fft(odd);

    // 组合结果
    for (int k = 0; k < N / 2; k++) {
        Complex t = polar(1.0, -2 * M_PI * k / N) * odd[k]; // 旋转因子 e^(-j2πk/N)
        x[k] = even[k] + t;
        x[k + N / 2] = even[k] - t;
    }
}

/**
 * @brief 递归实现逆快速傅里叶变换 (IFFT)
 * @param x 输入复数数组
 */
void ifft(CArray &x) {
    int N = x.size();
    if (N <= 1) return;

    // 拆分为偶数和奇数部分
    CArray even(N / 2), odd(N / 2);
    for (int i = 0; i < N / 2; i++) {
        even[i] = x[i * 2];
        odd[i] = x[i * 2 + 1];
    }

    // 递归计算 IFFT
    ifft(even);
    ifft(odd);

    // 组合结果
    for (int k = 0; k < N / 2; k++) {
        Complex t = polar(1.0, 2 * M_PI * k / N) * odd[k]; // 旋转因子 e^(j2πk/N)
        x[k] = (even[k] + t) / 2.0;
        x[k + N / 2] = (even[k] - t) / 2.0;
    }
}

/**
 * @brief 测试 FFT 和 IFFT
 */
int main() {
    CArray data = {1, 2, 3, 4, 5, 6, 7, 8}; // 输入信号

    cout << "原始数据:" << endl;
    for (const auto &val : data) cout << val << " ";
    cout << endl;

    fft(data); // 执行 FFT

    cout << "FFT 结果:" << endl;
    for (const auto &val : data) cout << val << " ";
    cout << endl;

    ifft(data); // 执行 IFFT

    cout << "逆 FFT 结果 (恢复信号):" << endl;
    for (const auto &val : data) cout << val << " ";
    cout << endl;

    return 0;
}