题解
看FFT原理的请移步这篇博客↓
P1919 【模板】A*B Problem升级版(FFT快速傅里叶)
c++自带的复数运用:
在输入复数的时候,发现复数只能输一个实部,
但是明明复数实部虚部两个值,为避免以后做到复数的题还要自己写operator,遂,想稍微扩展一下
第二个数输入需要另外的方法:
#include <bits/stdc++.h>
using namespace std;
int main() {
ios::sync_with_stdio(0);
int a, b, c, d;
cin >> a >> b >> c >> d; // 1 2 3 4
complex<double> x;
// cin>>x; 这么写就只是 x=(输入的实部,0)
x.real(a); // setReal 实部的输入方法 .real()
x.imag(b); // setImag 虚部的输入方法 .imag()
cout << x << endl; // (1,2)
complex<double> y;
y.real(c);
y.imag(d);
cout << y << endl; // (3,4)
cout << x + y << endl; // (4,6)
cout << x - y << endl; // (-2,-2)
cout << x * y << endl; // (-5,10)
cout << x / y << endl; // (0.44,0.08)
return 0;
}
#include <bits/stdc++.h>
using namespace std;
const int N = 3e6 + 10;//2e6+10 re
const double pi = acos(-1);
int n, m, k;
typedef complex<double> Complex;
int rev[N];
// FFT用于用于将系数表达式A转化为点值表达式
// inv=1 FFT inv=-1 FFT^-1
void FFT(Complex A[], int n, int inv) { //下标 0~n-1 n肯定是2的整数倍幂次 不足n位的 系数由0代替
/* 由于A B 两个函数长度不相同 所以最终的位置就必须在外部预处理
int bit = 0;
while ((1 << bit) < n) bit++;//找到最大的范围
//去递归
memset(rev, 0, sizeof(rev));*/
for (int i = 0; i < n; ++i) {
//rev[i] = (rev[i >> 1] >> 1) | ((i & 1) << (bit - 1));
if (i < rev[i]) swap(A[i], A[rev[i]]);
}
for (int mid = 1; mid < n; mid *= 2) { // mid=n/2
//wn1 = cos(1/n*2π)+i*sin(1/n*2π)
// = cos(1/(2mid)*2π)+i*sin(1/(2mid)*2π)
// = cos(π/mid)+i*sin(π/mid)
Complex wn = Complex(cos(pi / mid), inv * sin(pi / mid));
for (int i = 0; i < n; i += mid * 2) { // i表示合并到哪一位了
Complex w(1, 0);
for (int k = 0; k < mid; ++k, w *= wn) { //扫描左半部分 得到右半部分 w=wn^k
Complex x = A[i + k], y = w * A[i + k + mid];
A[i + k] = x + y;
A[i + k + mid] = x - y;
}
}
}
if (inv == -1) {
for (int i = 0; i < n; ++i) {
A[i] /= n;
}
}
}
int a, b;
Complex A[N], B[N];
int main() {
ios::sync_with_stdio(0);
cin >> a >> b;
for (int i = 0; i <= a; ++i) {
cin >> A[i]; // 输入的是实部 虚部为0
}
for (int i = 0; i <= b; ++i) {
cin >> B[i];
}
m = a + b + 1;// 函数最高次为a+b次 总共有 a+b+1 个系数
int bit = 0;
while ((1 << bit) < m) bit++;//找到最大的范围
n = (1 << bit);
//先预处理每个位置最终的位置
for (int i = 0; i < n; ++i) {
rev[i] = (rev[i >> 1] >> 1) | ((i & 1) << (bit - 1));
}
FFT(A, n, 1); //将系数表示的函数A 转化为 点值表示的函数A
FFT(B, n, 1); //将系数表示的函数B 转化为 点值表示的函数B
// A*B
for (int i = 0; i < n; ++i) {
A[i] *= B[i];
}
FFT(A, n, -1);//再将点值法转化为系数表达式
for (int i = 0; i < m; ++i) {
cout << (int) (A[i].real() + 0.5) << ' ';
}
return 0;
}