【模板】A*B Problem 升级版（FFT 快速傅里叶变换）

#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代替
    /* 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;
        }
    }
}

string a, b;
Complex A[N], B[N];

int ans[N];

int main() {
    ios::sync_with_stdio(0);
    //cin >> m;//得到的m可能不是2的整数倍幂次 需要将其转化为整数倍幂次 
    cin >> a >> b;
    n = a.length();
    m = b.length();
    //倒着存
    for (int i = 0; i < n; ++i) {
        A[i] = (double) (a[n - 1 - i] - '0'); // A[i]=<x,0>

    }
    for (int i = 0; i < m; ++i) {
        B[i] = (double) (b[m - 1 - i] - '0');
    }
    
    m = m + n - 1; // A(x)最高次为n-1次 B(x)最高次为m-1次 两个函数相乘最高次为n+m-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);
    FFT(B, n, 1);
    
    // A*B
    for (int i = 0; i < n; ++i) {
        A[i] *= B[i];
    }
    FFT(A, n, -1);//再将点值法转化为系数表达式

    //保存每一位答案 可能有进位 这里是倒着存的
    for (int i = 0; i < n; ++i) {
        //此时复数的虚部应该为0 或者 浮点误差接近0
        ans[i] += (int) (A[i].real() + 0.5);
        ans[i + 1] += ans[i] / 10;//进位
        ans[i] %= 10;
    }

    n--;
    while (!ans[n] && n) n--;//去掉因为FFT而补上的0
    for (int i = n; i >= 0; --i) {
        cout << ans[i];
    }
    return 0;
}