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

题目描述

给出两个 $n$ 位10进制数x和y，求x*y（详见 洛谷P1919）

分析

假设已经学会了FFT/NTT。

高精度乘法只是多项式乘法的特殊情况，相当于$x=10$ 时。

例如n=3，求123*111

$$123 = x^2 + 2x + 3$$

$$111 = x^2 + x +1$$

$$\begin{aligned}123 * 111 &= (x^2 + 2x + 3)(x^2 + x +1)\\  &= x^4 + 3x^3 + 6x^2 + 5x + 3\\  &= 13653\end{aligned}$$

代码：

#include<bits/stdc++.h>
#define rg register
using namespace std;

typedef long long ll;
const int mod=,g=;
const int maxn = 6e4 + ;

inline int qpow(int x,int k)
{
    int ans=;
    while(k)
    {
        if(k&)
            ans=(ll)ans*x%mod;
        x=(ll)x*x%mod,k>>=;
    }
    return ans;
}

inline int module(int x,int y)
{
    x+=y;
    if(x>=mod)
        x-=mod;
    return x;
}

int rev[*maxn];
inline void NTT(int*t,int lim,int type)
{
    for(rg int i=;i<lim;++i)
        if(i<rev[i])
            swap(t[i],t[rev[i]]);
    for(rg int i=;i<lim;i<<=)
    {
        int gn=qpow(g,(mod-)/(i<<));
        if(type==-)
            gn=qpow(gn,mod-);
        for(rg int j=;j<lim;j+=(i<<))
        {
            int gi=;
            for(rg int k=;k<i;++k,gi=(ll)gi*gn%mod)
            {
                int x=t[j+k],y=(ll)gi*t[j+i+k]%mod;
                t[j+k]=module(x,y);
                t[j+i+k]=module(x,mod-y);
            }
        }
    }
    if(type==-)
    {
        int inv=qpow(lim,mod-);
        for(rg int i=;i<lim;++i)
            t[i]=(ll)t[i]*inv%mod;
    }
}

int X[*maxn],Y[*maxn];
inline void mul(int*x, int*y, int n, int m)
{
    memset(X,,sizeof(X));
    memset(Y,,sizeof(Y));
    int lim = , L = ;  //L=0必须写，局部变量默认值很可能不是0
    while(lim <= n + m) lim <<= , L++;   //lim为大于(n+m)的2的幂，所以最多需要4倍空间
    for(int i = ; i < lim; i++) rev[i] = (rev[i >> ] >> ) | ((i & ) << (L - ));
    for(rg int i=;i<lim;++i) X[i]=x[i],Y[i]=y[i];
    NTT(X,lim,);
    NTT(Y,lim,);
    for(rg int i=;i<lim;++i) X[i]=(ll)X[i]*Y[i]%mod;
    NTT(X,lim,-);
    for(rg int i=;i<lim;++i) x[i]=X[i];
}

int n;
int a[*maxn], b[*maxn];
char s[maxn];

int main()
{
    scanf("%d", &n);
    scanf("%s", s);
    for(int i = ;i < n;i++) a[i] = s[n--i] - '';
    scanf("%s", s);
    for(int i = ;i < n;i++)  b[i] = s[n--i] - '';
    mul(a, b, n, n);

//    for(int i = 0;i < 2*n;i++)  printf("%d ", a[i]);
//    printf("\n");

    int tmp = ;    //进位
    for(int i = ;i < *n;i++)  //
    {
        a[i] = a[i] + tmp;
        tmp = a[i] / ;
        a[i] = a[i] % ;

    }

//    for(int i = 0;i < 2*n;i++)  printf("%d ", a[i]);
//    printf("\n");

    bool flag = true;
    for(int i = *n;i >= ;i--)  //逆序输出，去掉前导零
    {
        if(flag && a[i] == )  continue;
         printf("%d", a[i]);
         flag = false;
    }

    return ;
}