快速傅里叶变换FFT实现的c++代码:
#include <iostream>
using namespace std;
#include <complex>
#include <cmath>
double PI = acos(-1);
complex<double> a[400010], b[400010], c[400010];
void fft(complex<double>* a, int n, int op)
{
if (n == 1) return;
complex<double> w(1, 0), wn(cos(2 * PI * op / n), sin(2 * PI * op / n)), a1[10], a2[10];
for (int i = 0; i < (n >> 1); i++)
a1[i] = a[i << 1], a2[i] = a[(i << 1) + 1];
fft(a1, n >> 1, op), fft(a2, n >> 1, op);
for (int i = 0; i < (n >> 1); i++, w *= wn)
a[i] = a1[i] + w * a2[i], a[i + (n >> 1)] = a1[i] - w * a2[i];
}
int main()
{
int n, m;
scanf("%d%d", &n, &m);
for (int i = 0; i <= n; i++) scanf("%lf", &a[i]);
for (int i = 0; i <= m; i++) scanf("%lf", &b[i]);
m += n, n = 1;
while (n <= m) n <<= 1;
fft(a, n, 1), fft(b, n, 1);
for (int i = 0; i < n; i++) c[i] = a[i] * b[i];
fft(c, n, -1);
for (int i = 0; i <= m; i++) printf("%d ", int(c[i].real() / n + 0.5));
return 0;
}
VS2019不允许使用scanf()函数,觉得修改比较麻烦的话,可试试这样操作,保留scanf()函数而不报错:
在VS界面的左上菜单栏的项目——选择你的项目属性(应该是最后一个)——C/C++——选择预处理器——点击编辑——添加**_CRT_SECURE_NO_WARNINGS**
参考博文中还有另外的方法实现,从算法的角度应该更快。
参考博文:
FFT:https://blog.csdn.net/xs1997/article/details/78175559
VS中修改预处理器:https://blog.csdn.net/SartinL/article/details/104050851