首页 > 代码库 > 【FFT】hdu1402 A * B Problem Plus
【FFT】hdu1402 A * B Problem Plus
FFT板子。
将大整数看作多项式,它们的乘积即多项式的乘积在x=10处的取值。
#include<cstdio> #include<cmath> #include<cstring> #include<algorithm> using namespace std; #define EPS 1e-8 const double PI = acos(-1.0); struct Complex{ double real,image; Complex(double _real,double _image){ real=_real; image=_image; } Complex(){} }; Complex operator + (const Complex &c1,const Complex &c2){ return Complex(c1.real+c2.real,c1.image+c2.image); } Complex operator - (const Complex &c1,const Complex &c2){ return Complex(c1.real-c2.real,c1.image-c2.image); } Complex operator * (const Complex &c1,const Complex &c2){ return Complex(c1.real*c2.real-c1.image*c2.image,c1.real*c2.image+c1.image*c2.real); } int rev(int id,int len){ int ret=0; for(int i=0;(1<<i)<len;++i){ ret<<=1; if(id&(1<<i)){ ret|=1; } } return ret; } Complex tmp[200100]; //μ±DFT==1ê±ê?DFT, DFT==-1ê±?òê???DFT void IterativeFFT(Complex A[],int len, int DFT){//??3¤?è?alen(2μ??Y)μ?êy×é??DDDFT±??? for(int i=0;i<len;++i){ tmp[rev(i,len)]=A[i]; } for(int i=0;i<len;++i){ A[i]=tmp[i]; } for(int s=1;(1<<s)<=len;++s){ int m=(1<<s); Complex wm=Complex(cos(DFT*2*PI/m),sin(DFT*2*PI/m)); for(int k=0;k<len;k+=m){//?aò?2??áμ?°üo?μ?êy×é?a????êy??ê?(1<<s) Complex w=Complex(1,0); for(int j=0;j<(m>>1);++j){//??°?òyàí£??ù?Yá???×ó?úμ????????úμ? Complex t=w*A[k+j+(m>>1)]; Complex u=A[k+j]; A[k+j]=u+t; A[k+j+(m>>1)]=u-t; w=w*wm; } } } if(DFT==-1){ for(int i=0;i<len;++i){ A[i].real/=len; A[i].image/=len; } } } Complex a[200100],b[200100]; int ans[200100]; char s1[50010],s2[50010]; int main(){ // freopen("a.in","r",stdin); while(scanf("%s%s",s1,s2)!=EOF){ memset(a,0,sizeof(a)); memset(b,0,sizeof(b)); memset(ans,0,sizeof(ans)); int len1=strlen(s1),len2=strlen(s2),len; if((len1==1 && s1[0]==‘0‘) || (len2==1 && s2[0]==‘0‘)){ puts("0"); continue; } for(int i=0;;++i){ if((1<<i)>=len1+len2){ len=(1<<i); break; } } for(int i=len1-1,j=0;i>=0;--i,++j){ a[j]=Complex(s1[i]-‘0‘,0); } for(int i=len2-1,j=0;i>=0;--i,++j){ b[j]=Complex(s2[i]-‘0‘,0); } IterativeFFT(a,len,1); IterativeFFT(b,len,1); for(int i=0;i<len;++i){ a[i]=a[i]*b[i]; } IterativeFFT(a,len,-1); for(int i=0;i<len;++i){ ans[i]=(int)(a[i].real+0.5); } for(int i=0;i<len1+len2-1;++i){ ans[i+1]+=ans[i]/10; ans[i]%=10; } for(int i=len;i>=0;--i){ if(ans[i]!=0){ for(int j=i;j>=0;--j){ printf("%d",ans[j]); } puts(""); break; } } } return 0; }
【FFT】hdu1402 A * B Problem Plus
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。