zoukankan      html  css  js  c++  java
  • luogu P1919 【模板】A*B Problem升级版(FFT快速傅里叶)|FFT

    题目描述

    给你两个正整数 a,b,求 a×b。

    输入格式

    第一行一个正整数,表示 a;
    第二行一个正整数,表示 b。

    输出格式

    输出一行一个整数表示答案。

    #include<cmath>
    #include<string>  
    #include<cstdio>
    #include<iostream>
    #include<algorithm>
    using namespace std;
    const int _=4e6+10;
    const double Pi=acos(-1.0);
    #define db double
    struct lex{
    	db x,y;
    	lex(db _x=0,db _y=0){x=_x;y=_y;}
    	lex operator + (const lex &b)const{return lex(x+b.x,y+b.y);}
    	lex operator - (const lex &b)const{return lex(x-b.x,y-b.y);}
    	lex operator * (const lex &b)const{return lex(x*b.x-y*b.y,x*b.y+y*b.x);}
    }a[_],b[_];
    int N,M;
    inline void fft(int lim,lex *a,int op){
    	if(lim==1)return;
    	lex a1[(lim>>1)+5],a2[(lim>>1)+5];
    	for(int i=0;i<lim;i+=2)a1[i>>1]=a[i],a2[i>>1]=a[i+1];
    	fft(lim>>1,a1,op);
    	fft(lim>>1,a2,op);
    	lex Wn=lex(cos(2.0*Pi/lim),op*sin(2.0*Pi/lim));
    	lex w=lex(1,0);
    	for(int i=0;i<(lim>>1);i++,w=w*Wn){
    		a[i]=a1[i]+w*a2[i];
    		a[i+(lim>>1)]=a1[i]-w*a2[i];
    	}
    }
    int ans[_],AA,BB;
    signed main(){
    	string s1,s2; cin>>s1>>s2;
    	N=s1.length(); M=s2.length();
    	register int i;
    	for(i=N-1;i>=0;i--)a[AA++].x=s1[i]-'0';
    	for(i=M-1;i>=0;i--)b[BB++].x=s2[i]-'0';
    	register int lim=1; while(lim<=N+M)lim<<=1;
    	fft(lim,a,1);
    	fft(lim,b,1);
    	for(i=0;i<=lim;i++)a[i]=a[i]*b[i];
    	fft(lim,a,-1);
    	for(i = 0; i <= lim; i++) {
            ans[i]+=(int)(a[i].x/lim+0.5);
            if(ans[i]>=10)ans[i + 1]+=ans[i]/10,ans[i]%=10,lim+=(i==lim);
        }
        while(!ans[lim] && lim >= 1) lim -- ;
        lim ++ ;
        while(--lim>= 0)printf("%d",ans[lim]);
        return 0 ;
    }
    
  • 相关阅读:
    MyBatis
    spring post 图片
    spring boot
    linux 调用栈打印
    sys.usb.config webcam
    jsp+bean+servlet 案例代码
    常见的http头信息
    通过JSTL+EL实现循环迭代
    EL表达式介绍
    EL表达式
  • 原文地址:https://www.cnblogs.com/naruto-mzx/p/12106243.html
Copyright © 2011-2022 走看看