BZOJ2179: FFT快速傅立叶 FFT实现高精度乘法

Description

给出两个n位10进制整数x和y，你需要计算x*y。

Input

第一行一个正整数n。 第二行描述一个位数为n的正整数x。 第三行描述一个位数为n的正整数y。

Output

输出一行，即x*y的结果。

题解： 一个数字可以被写成 $\sum a_{i}\times10^i$ 的形式，那么两个数字相乘就是 $(\sum a_{i}\times 10^i)\times(\sum b_{i}\times 10^i)$，这可以看成两个多项式相乘的形式，可以用 $FFT$ 来加速乘法
有一些细节需要注意：$a$ 和 $b$ 的 $0$ 位要对应个位，依次类推，注意进位

#include <cstdio>
#include <algorithm>
#include <cmath>
#include <cstring>
#define setIO(s) freopen(s".in","r",stdin)
#define maxn 200000
#define pi 3.1415926535898
using namespace std;
int len=1,l,r[maxn<<1];
long long ans[maxn];
char str1[maxn],str2[maxn];
struct Cpx{
    double x,y;
    Cpx(double t1=0,double t2=0){x=t1,y=t2;}
}A[maxn<<1],B[maxn<<1],C[maxn<<1];
Cpx operator+(Cpx a,Cpx b){ return Cpx(a.x+b.x,a.y+b.y);}
Cpx operator-(Cpx a,Cpx b){ return Cpx(a.x-b.x,a.y-b.y);}
Cpx operator*(Cpx a,Cpx b){ return Cpx(a.x*b.x-a.y*b.y,a.x*b.y+a.y*b.x);}
void FFT(Cpx *a,int n,int flag){
    for(int i=0;i<n;++i) if(i<r[i]) swap(a[i],a[r[i]]);
    for(int mid=1;mid<n;mid<<=1){
        Cpx wn(cos(pi/mid),flag*sin(pi/mid)),x,y;
        for(int j=0;j<n;j+=(mid<<1)){
            Cpx w(1,0);
            for(int k=0;k<mid;++k){
                x=a[j+k],y=w*a[j+mid+k];
                a[j+k]=x+y,a[j+mid+k]=x-y;
                w=w*wn;
            }
        }
    }
}
int main(){
    //setIO("input");
    int n;
    scanf("%d",&n);
    scanf("%s%s",str1,str2);
    for(int i=0;i<n;++i) A[i].x=str1[n-i-1]-48;
    for(int i=0;i<n;++i) B[i].x=str2[n-i-1]-48;
    while(len<n+n) len<<=1,++l;
    for(int i=0;i<len;++i)
        r[i]=(r[i>>1]>>1)|((i&1)<<(l-1));
    FFT(A,len,1),FFT(B,len,1);
    for(int i=0;i<len;++i)
        C[i]=A[i]*B[i];
    FFT(C,len,-1);
    for(int i=0;i<=len;++i) {
        ans[i]+=(int)(C[i].x/len+0.5);
        if(ans[i]>=10)
            ans[i+1]+=ans[i]/10,ans[i]%=10,
        len+=(len==i);
    }
    while(!ans[len]&&len>=0) --len;
    while(len>=0) printf("%d",(int)ans[len]),--len;
    return 0;
}