P1919 【模板】A*B Problem升级版(FFT快速傅里叶)
分析:
为什么高精乘可以用 FFT 优化???
众所周知 FFT 是解决多项式之间的乘法运算,将乘数的每一位看成多项式的系数即可
#include<bits/stdc++.h>
using namespace std;
const int N = (1<<21)+5;
const double PI=acos(-1);
struct Complex
{
double x, y;
Complex operator+(const Complex &o) const{return{x+o.x,y+o.y};}
Complex operator-(const Complex &o) const{return{x-o.x,y-o.y};}
Complex operator*(const Complex &o) const{return{x*o.x-y*o.y,x*o.y+y*o.x};}
}A[N],B[N];
int rev[N];
void init(int k)
{
int s=1<<k;
for(int i=1;i<s;i++) rev[i]=(rev[i>>1]>>1)|((i&1)<<(k-1));
}
void fft(Complex *a,int n,int inv)
{
for(int i=0;i<n;i++) if(i<rev[i]) swap(a[i],a[rev[i]]);
for(int len=1;len<n;len<<=1)
{
Complex Wn=Complex({cos(PI/len),inv*sin(PI/len)});
for(int i=0;i<n;i+=len*2)
{
Complex w=Complex({1,0});
for(int j=0;j<len;j++,w=w*Wn)
{
Complex x=a[i+j],y=w*a[i+j+len];
a[i+j]=x+y,a[i+j+len]=x-y;
}
}
}
if(inv==-1) for(int i=0;i<n;i++) A[i].x = A[i].x/n+0.5;
}
// --------FFT
char s1[N], s2[N];
int ans[N];
int main()
{
scanf("%s%s",s1,s2);
int n=strlen(s1), m=strlen(s2);
for(int i=0;i<n;i++)A[i].x=(double)(s1[n-i-1]-'0');
for(int i=0;i<m;i++)B[i].x=(double)(s2[m-i-1]-'0');
int k=1,s=2;
while(s < n+m-1)k++,s<<=1;
init(k);
fft(A,s,1); fft(B,s,1);
for(int i=0;i<s;i++) A[i] = A[i]*B[i];
fft(A,s,-1);
for(int i=0;i<s;i++)
{
ans[i]+=(int)(A[i].x);
ans[i+1]+=ans[i]/10;
ans[i]%=10;
}
while(!ans[s] &&s >-1) s--;
if(s==-1) cout<<"0";
else {
for(int i=s;i>=0;i--) cout<<ans[i];
}
return 0;
}