P3803 【模板】多项式乘法（FFT）

题解

看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;
}