【CodeVS 3123】 高精度练习之超大整数乘法

RE了2发 init竟然会开小.......

1 #include <cstdio>
2 #include <algorithm>
3 #include <complex>
4 #include <iostream>
5 #include <cmath>
6 using namespace std;
7 const double PI= acos(-1);
8 int init[100000*4+10];
9 #define C complex<double>
C a[100000*4+1],b[100000*4+1];
int n,m,rev[100000*4];
void FFT(C a[],int type)
{
for(int i=0;i<n;i++) if(rev[i]<i) swap(a[rev[i]],a[i]);
for(int i=2;i<=n;i<<=1)
{
C wn(cos(2*PI/i),type*sin(2*PI/i));
for(int j=0;j<n;j+=i)
{
C w(1,0),x,y;
for(int k=0;k<i/2;k++)
{
x=a[j+k];
y=a[j+k+i/2]*w;
a[j+k]=x+y;
a[j+k+i/2]=x-y;
w=w*wn;
}
}
}
}
int main()
{
//freop0en("a.in","r",stdin);
char c=getchar();
while(c!=' ') init[++n]=c-'0',c=getchar();
for(int i=n;i>=1;i--) a[n-i]=init[i];
c=getchar();
while(c!='\n') init[++m]=c-'0',c=getchar();
for(int i=m;i>=1;i--) b[m-i]=init[i];

n--,m--;
// for(int i=0;i<=n;i++) cout<<a[i]<<' ';cout<<endl;
// for(int i=0;i<=m;i++) cout<<b[i]<<' ';cout<<endl;
m=n+m;int L=0;
for(n=1;n<=m;n<<=1)L++; L--;
for(int i=0;i<n;i++) rev[i]=((rev[i>>1]>>1)|((i&1)<<L));

FFT(a,1);FFT(b,1);
for(int i=0;i<n;i++) a[i]*=b[i];
FFT(a,-1);
for(int i=0;i<=m;i++) init[i]=(int)(a[i].real()/n+0.5);//,cout<<init[i]<<' ';cout<<endl;
init[m+1]=0;
for(int i=0;i<=m;i++) init[i+1]+=init[i]/10,init[i]%=10;//,cout<<init[i+1]<<' ';cout<<endl;
if(init[m+1]!=0) m++;
for(int i=m;i>=0;i--) printf("%d",init[i]);
return 0;
}