FFT高精度乘法

众所周知,FFT(Fast Fourier Transformation),即为快速傅氏变换,是离散傅氏变换的快速算法,它是根据离散傅氏变换的奇、偶、虚、实等特性,对离散傅立叶变换的算法进行改进获得的.
那么我们用这个nb的算法解决一些什么沙雕问题呢QwQ???
没错!就是大名鼎鼎的高精度乘法
FFT太难了我不会啊QAQ,NTT更难了哇怎么办啊

行吧,上代码

#include<algorithm>
#include<cstring>
#include<cstdio>
#include<cmath>
#define N 200000
using namespace std;
const double DFT=2.0,IDFT=-2.0;
const double pi=acos(-1),eps=0.5;
struct Complex
{
    double x,y;
    Complex(double _=0,double __=0):x(_),y(__){}
    Complex operator + (const Complex &b) const{return Complex(x+b.x,y+b.y);}
    Complex operator - (const Complex &b) const{return Complex(x-b.x,y-b.y);}
    Complex operator * (const Complex &b) const{return Complex(x*b.x-y*b.y,x*b.y+y*b.x);}
}a[N],b[N],c[N];
char s1[N],s2[N];
int pos[N],n,top,len,ans[N];
inline void FFT(Complex *a,double mode)
{
    for(int i=0;i<len;i++)if(i<pos[i])swap(a[i],a[pos[i]]);
    for(int i=2,mid=1;i<=len;i<<=1,mid<<=1)
    {
        Complex wm(cos(2.0*pi/i),sin(mode*pi/i));
        for(int j=0;j<len;j+=i)
        {
            Complex w(1.0,0.0);
            for(int k=j;k<j+mid;k++,w=w*wm)
            {
                Complex l=a[k],r=w*a[k+mid];
                a[k]=l+r;a[k+mid]=l-r;
            }
        }
    }
    if(mode==IDFT) for(int i=0;i<len;i++) a[i].x/=len;
    return;
}
int main()
{
    scanf("%d%s%s",&n,s1,s2);
    for(len=1;len<=n<<1;len<<=1);
    for(int i=0;i<len;i++)
    {
        pos[i]=pos[i>>1]>>1;
        if(i&1) pos[i]|=len>>1;
    }
    int o=0;
    for(o=0;o<n;o++){a[o].x=s1[n-o-1]-'0';a[o].y=0.0;}
    while(o<len){a[o].x=0;a[o++].y=0;}
    for(o=0;o<n;o++){b[o].x=s2[n-o-1]-'0';b[o].y=0.0;}
    while(o<len){b[o].x=0;b[o++].y=0;}
    FFT(a,DFT);FFT(b,DFT);
    for(int i=0;i<len;i++) c[i]=a[i]*b[i];
    FFT(c,IDFT);
    for(int i=0;i<len;i++) ans[i]=int(c[i].x+0.5);
    for(int i=0;i<len;i++)
    {
        ans[i+1]+=ans[i]/10;
        ans[i]%=10;
    }
    for(int i=len-1;i>=0;i--) if(ans[i]){top=i;break;}
    for(int i=top;i>=0;i--) printf("%d",ans[i]);
    return 0;
}
  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值