C语言实现离散傅里叶变换 (DFT) 并行优化 - OpenMP

#define _USE_MATH_DEFINES\n#include <stdio.h>\n#include <stdlib.h>\n#include <math.h>\n#include <omp.h>\n\n#define N 1024\n\nvoid dft(double *in, double *out, int size)\n{\n\t\tint i, j;\n\t\tfor (i = 0; i < size; ++i)\n\t\t{\n\t\t\tout[i] = 0.0;\n\t\t\tfor (j = 0; j < size; ++j)\n\t\t\t{\n\t\t\t\tdouble angle = 2 * M_PI * i * j / size;\n\t\t\t\tout[i] += in[j] * cos(angle) - in[j] * sin(angle);\n\t\t\t}\n\t\t}\n}\n\nvoid parallel_dft(double *in, double *out, int size)\n{\n\t\tint i, j;\n#pragma omp parallel for private(j)\n\t\tfor (i = 0; i < size; ++i)\n\t\t{\n\t\t\tout[i] = 0.0;\n\t\t\tfor (j = 0; j < size; ++j)\n\t\t\t{\n\t\t\t\tdouble angle = 2 * M_PI * i * j / size;\n\t\t\t\tout[i] += in[j] * cos(angle) - in[j] * sin(angle);\n\t\t\t}\n\t\t}\n}\n\nint main()\n{\n\t\tdouble input[N];\n\t\tdouble output[N];\n\t\tint i;\n\n\t\t// Initialize input data\n\t\tfor (i = 0; i < N; ++i)\n\t\t{\n\t\t\tinput[i] = i;\n\t\t}\n\n\t\t// Perform DFT using serial implementation\n\t\tdft(input, output, N);\n\t\tprintf("Serial DFT: Output[0] = %f, Output[1] = %f\n", output[0], output[1]);\n\n\t\t// Perform DFT using parallel implementation\n\t\tparallel_dft(input, output, N);\n\t\tprintf("Parallel DFT: Output[0] = %f, Output[1] = %f\n", output[0], output[1]);\n\n\t\treturn 0;\n}\n这段代码的设计思想是实现离散傅里叶变换（DFT）的串行和并行版本。DFT是一种将时域信号转换为频域信号的数学变换，可以用于信号处理、图像处理等领域。串行版本通过嵌套循环遍历输入数组中的元素，并根据DFT公式计算输出数组中的元素。并行版本使用OpenMP库中的并行循环指令（pragma omp parallel for）将外层循环并行化，以加快计算速度和利用多核处理器的并行能力。代码中的输入数组input包含了0到N-1的连续整数，用于模拟一个时域信号。最后，代码分别输出了串行和并行版本的计算结果。