NKOJ 3071 【高精度】a*b
问题描述
给你两个正整数a,b,计算它们的乘积。
输入格式
第一行一个正整数a
第二行一个正整数b
输出格式
一行,表示a*b
样例输入
111222333444555666777888999
999888777666555444333222111
样例输出
111209963037098814851876554444456814851901296370456889
提示
a,b分别不超过100000位
FFT裸题。由于是第一道FFT所以纪念一下。代码总算是背下来了。
注意FFT完之后可能出现精度问题和大小超过10的情况,关注处理进位的方法即可。
#include<stdio.h>
#include<cmath>
#include<complex>
#include<cstring>
#define MAXN 412345
using namespace std;
const double Pi=3.1415926535897932384626433832795;
complex<double>fft[MAXN];
char a[MAXN],b[MAXN];
complex<double>A[MAXN],B[MAXN];
int Ans[MAXN];
void FFT(complex<double>A[],int n,int ty)
{
int i,j,k,m;
complex<double>t0,t1;
for(i=0;i<n;i++)
{
for(j=0,k=i,m=1;m<n;m<<=1,j=(j<<1)|(k&1),k>>=1);
if(i<j)t0=A[i],A[i]=A[j],A[j]=t0;
}
fft[0]=1;
for(m=1;m<n;m<<=1)
{
t0=exp(complex<double>(0,ty*Pi/m));
for(i=1;i<m;i++)fft[i]=fft[i-1]*t0;
for(k=0;k<n;k+=m<<1)
for(i=k;i<k+m;i++)
{
t0=A[i];
t1=A[i+m]*fft[i-k];
A[i]=t0+t1;
A[i+m]=t0-t1;
}
}
if(ty==1)return;
t0=1.0/n;
for(i=0;i<n;i++)A[i]*=t0;
}
int main()
{
int i,j,k,la,lb,N=1;
scanf("%s",a);
scanf("%s",b);
la=strlen(a);
lb=strlen(b);
for(i=0;i<la;i++)A[la-i-1]=a[i]-48;
for(i=0;i<lb;i++)B[lb-i-1]=b[i]-48;
while(N<=la+lb+1)N<<=1;
FFT(A,N,1);FFT(B,N,1);
for(i=0;i<N;i++)A[i]*=B[i];
FFT(A,N,-1);
for(i=0;i<N;i++)Ans[i]=floor(A[i].real()+0.5);
for(i=0;i<N;i++)Ans[i+1]+=Ans[i]/10,Ans[i]%=10;
i=N;while(!Ans[i])i--;
while(i>=0)printf("%d",Ans[i]),i--;
}