hdu 1402 A * B Problem Plus(FFT)

题目链接：hdu 1402 A * B Problem Plus

题意：

让你求两个大数的乘法。

题解：

FFT裸题。

FFT学习推荐：FFT学习笔记

#include<bits/stdc++.h>
#define F(i,a,b) for(int i=a;i<=b;++i)
using namespace std;

const double pi=acos(-1.0);
//n 必须为 2 的幂。
struct comp{
    double r,i;
    comp(double _r=0,double _i=0){r=_r,i=_i;}
    comp operator+(const comp x){return comp(r+x.r,i+x.i);}
    comp operator-(const comp x){return comp(r-x.r,i-x.i);}
    comp operator*(const comp x){return comp(r*x.r-i*x.i,r*x.i+i*x.r);}
};

void FFT(comp a[],int n,int t){
    for(int i=1,j=0;i<n-1;i++)
    {
        for(int s=n;j^=s>>=1,~j&s;);
        if(i<j)swap(a[i],a[j]);
    }
    for(int d=0;(1<<d)<n;d++)
    {
        int m=1<<d,m2=m<<1;
        double o=pi/m*t;comp _w(cos(o),sin(o));
        for(int i=0;i<n;i+=m2)
        {
            comp w(1,0);
            for(int j=0;j<m;j++)
            {
                comp &A=a[i+j+m],&B=a[i+j],t=w*A;
                A=B-t,B=B+t,w=w*_w;
            }
        }
    }
    if(t==-1)for(int i=0;i<n;i++)a[i].r/=n;
}

const int N=2e5+7;

comp x1[N],x2[N];
char a[N],b[N];
int sum[N];

int main()
{
    while(~scanf("%s%s",a,b))
    {
        int lena=strlen(a),lenb=strlen(b);
        int n=1;
        while(n<lena*2||n<lenb*2)n<<=1;
        F(i,0,n-1)
        {
            if(i<lena)x1[i]=comp(a[lena-1-i]-'0',0);
            else x1[i]=comp(0,0);
        }
        F(i,0,n-1)
        {
            if(i<lenb)x2[i]=comp(b[lenb-1-i]-'0',0);
            else x2[i]=comp(0,0);
        }
        FFT(x1,n,1),FFT(x2,n,1);
        F(i,0,n-1)x1[i]=x1[i]*x2[i];
        FFT(x1,n,-1);
        F(i,0,n-1)sum[i]=(int)(x1[i].r+0.5);
        F(i,0,n-1)
        {
            sum[i+1]+=sum[i]/10;
            sum[i]%=10;
        }
        n--;
        while(sum[n]<=0&&n>0)n--;
        for(int i=n;i>=0;i--)printf("%c",sum[i]+'0');
        puts("");
    }
    return 0;
}