HDU1402 A * B Problem Plus FFT

时间:2023-03-09 13:33:24
HDU1402 A * B Problem Plus  FFT

分析:网上别家的代码都分析的很好,我只是给我自己贴个代码,我是kuangbin的搬运工

一点想法:其实FFT就是快速求卷积罢了,当小数据的时候我们完全可以用母函数来做,比如那种硬币问题

FFT只是用来解决数据规模较大时的办法,可以达到nlogn的效率,大体原理就是运用了n次单位复根的折半引理

具体可以看算法导论

高度仰慕kuangbin大神的模板,实在是不想自己写

对于这个题,我们10^k的系数看成多项式系数,然后求卷积,进位就好了

#include <stdio.h>
#include <iostream>
#include <algorithm>
#include <string.h>
#include <vector>
#include <math.h>
#include <map>
using namespace std;
typedef long long LL;
const int INF=0x3f3f3f3f;
const int N=2e5+;
const double pi = acos(-1.0);
struct complex{
double r,i;
complex(double R=,double I=){
r=R;i=I;
}
complex operator+(const complex &a)const{
return complex(r+a.r,i+a.i);
}
complex operator-(const complex &a)const{
return complex(r-a.r,i-a.i);
}
complex operator*(const complex &a)const{
return complex(r*a.r-i*a.i,r*a.i+i*a.r);
}
};
void change(complex x[],int len){
int i,j,k;
for(i=,j=len/;i<len-;++i){
if(i<j)swap(x[i],x[j]);
k=len/;
while(j>=k){j-=k;k>>=;}
if(j<k)j+=k;
}
}
void fft(complex x[],int len,int on){
change(x,len);
for(int i=;i<=len;i<<=){
complex wn(cos(-on**pi/i),sin(-on**pi/i));
for(int j=;j<len;j+=i){
complex w(,);
for(int k=j;k<j+i/;++k){
complex u = x[k];
complex t = w*x[k+i/];
x[k]=u+t;
x[k+i/]=u-t;
w=w*wn;
}
}
}
if(on==-)for(int i=;i<len;++i)x[i].r/=len;
}
complex x1[N],x2[N];
char str1[N/],str2[N/];
int sum[N];
int main(){
while(~scanf("%s%s",str1,str2)){
int len1 = strlen(str1);
int len2 = strlen(str2),len=;
while(len<len1*||len<len2*)len<<=;
for(int i=;i<len1;++i)
x1[i]=complex(str1[len1--i]-'',);
for(int i=len1;i<len;++i)
x1[i]=complex(,);
for(int i=;i<len2;++i)
x2[i]=complex(str2[len2--i]-'',);
for(int i=len2;i<len;++i)
x2[i]=complex(,);
fft(x1,len,);
fft(x2,len,);
for(int i=;i<len;++i)
x1[i]=x1[i]*x2[i];
fft(x1,len,-);
for(int i=;i<len;++i)
sum[i]=(int)(x1[i].r+0.5);
for(int i=;i<len;++i){
sum[i+]+=sum[i]/;
sum[i]%=;
}
len = len1+len2-;
while(sum[len]<=&&len>)--len;
for(int i=len;i>=;--i)printf("%d",sum[i]);
printf("\n");
}
return ;
}