HDU 1402 A * B Problem Plus 快速傅里叶变换 FFT 多项式

http://acm.hdu.edu.cn/showproblem.php?pid=1402

快速傅里叶变换优化的高精度乘法。

https://blog.csdn.net/ggn_2015/article/details/68922404 这个写的很详细了。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<iostream>
#include<vector>
#include<complex>
using namespace std;
#define LL long long
const int maxn=150010;
typedef complex< double >cd;
char s1[maxn]={},s2[maxn]={};
int ans[maxn]={};
int rev[maxn]={}; int s,bit;
cd a[maxn]={},b[maxn]={};
double Pi;
inline void getrev(){ for(int i=0;i<s;i++)rev[i]=(rev[i>>1]>>1)|((i&1)<<(bit-1)); }
inline void fft(cd *c,int n,int dft){
    for(int i=1;i<=s;i++)if(rev[i]>i)swap(c[i],c[rev[i]]);
    for(int step=1;step<n;step<<=1){
        cd shu=exp(cd(0,dft*Pi/step));
        for(int i=0;i<n;i+=step<<1){
            cd z=cd(1,0); 
            for(int j=i;j<i+step;j++){
                cd x=c[j];cd y=c[j+step]*z;
                c[j]=x+y; c[j+step]=x-y;
                z*=shu;
            }
        }
    }
    if(dft==-1)for(int i=0;i<n;i++)c[i]/=n;
}
int main(){
    cd cle(0,0);Pi=2.0*acos(0.0);
    while(~scanf("%s%s",s1,s2)){
        memset(rev,0,sizeof(rev));
        int l1=strlen(s1),l2=strlen(s2),n=l1+l2-1;
        bit=1;s=2; for(;s<n;++bit)s<<=1;
        getrev();
        for(int i=1;i<=s;i++){a[i]=cle;b[i]=cle;}
        for(int i=0;i<l1;i++)a[i]=(double)(s1[l1-i-1]-'0');
        for(int i=0;i<l2;i++)b[i]=(double)(s2[l2-i-1]-'0');
        fft(a,s,1);fft(b,s,1);
        for(int i=0;i<s;i++)a[i]*=b[i];
        fft(a,s,-1);
        memset(ans,0,sizeof(ans));
        for(int i=0;i<s;i++){
            ans[i]+=(int)(a[i].real()+0.5);
            ans[i+1]+=ans[i]/10;
            ans[i]%=10;
        }
        int i;
        for(i=l1+l2;!ans[i]&&i>0;--i);
        if(i==0)printf("%d
",ans[0]);
        else{
            for(;i>=0;--i)printf("%d",ans[i]);
            printf("
");
        }
    }
    return 0;
}