【bzoj2179】FFT快速傅立叶 FFT模板

Description

给出两个n位10进制整数x和y,你需要计算x*y。
Input

第一行一个正整数n。 第二行描述一个位数为n的正整数x。 第三行描述一个位数为n的正整数y。
Output

输出一行,即x*y的结果。
Sample Input

1

3

4

Sample Output

12

数据范围:

n<=60000

题解
FFT模板(暂时只会递归版)
http://www.gatevin.moe/acm/fft%E7%AE%97%E6%B3%95%E5%AD%A6%E4%B9%A0%E7%AC%94%E8%AE%B0/ (里面大概就是算导的内容)

代码(递归版)

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<complex>
#define ll long long
#define mod 10000007
using namespace std;
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if (ch=='-') f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
typedef complex<double> E;
#define N 200000
#define pi acos(-1)
int n,ans[N],m;
E a[N],b[N];
char ch[N];
void FFT(E *a,int n,int f)
{
    if (n==1) return;
    E l[n>>1],r[n>>1];
    for (int i=0;i<n;i+=2) l[i>>1]=a[i],r[i>>1]=a[i+1];
    FFT(l,n>>1,f);FFT(r,n>>1,f);
    E wn(cos(2*pi/n),f*sin(2*pi/n)),w(1,0),t;
    for (int i=0;i<(n>>1);i++,w*=wn)
    {
        t=w*r[i];
        a[i]=l[i]+t;
        a[i+(n>>1)]=l[i]-t;
    }
}
int main()
{
    n=read();n--;
    scanf("%s",ch);
    for(int i=0;i<=n;i++)a[i]=ch[n-i]-'0';
    scanf("%s",ch);
    for(int i=0;i<=n;i++)b[i]=ch[n-i]-'0';
    m=2*n;for (n=1;n<=m;n<<=1);
    FFT(a,n,1);FFT(b,n,1);
    for (int i=0;i<n;i++) a[i]*=b[i];
    FFT(a,n,-1);
    for (int i=0;i<n;++i) ans[i]=(int)(a[i].real()/n+0.5);
    for (int i=0;i<n;++i) if (ans[i]>9) ans[i+1]+=ans[i]/10,ans[i]%=10;
    while (!ans[n]) n--;
    for (int i=n;i>=0;i--) printf("%d",ans[i]);cout<<endl;
}

补:非递归版

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<complex>
#define ll long long
#define mod 10000007
using namespace std;
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if (ch=='-') f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
typedef complex<double> E;
#define N 200000
#define pi acos(-1)
int n,ans[N],m,R[N],L;
E a[N],b[N];
char ch[N];
void FFT(E *a,int f)
{
    for (int i=0;i<n;i++) if (i<R[i]) swap(a[i],a[R[i]]);
    for (int i=1;i<n;i<<=1)
    {
        E wn(cos(pi/i),f*sin(pi/i));
        for (int j=0;j<n;j+=(i<<1))
        {
            E w(1,0);
            for (int k=0;k<i;k++,w*=wn)
            {
                E x=a[j+k],y=w*a[j+k+i];
                a[j+k]=x+y;a[j+k+i]=x-y;
            }
        }
    }
}
int main()
{
    n=read();n--;
    scanf("%s",ch);
    for(int i=0;i<=n;i++)a[i]=ch[n-i]-'0';
    scanf("%s",ch);
    for(int i=0;i<=n;i++)b[i]=ch[n-i]-'0';
    m=2*n;for (n=1;n<=m;n<<=1) L++;
    for (int i=0;i<n;i++) R[i]=(R[i>>1]>>1)|((i&1)<<(L-1)); //其实这句还是不懂
    FFT(a,1);FFT(b,1);
    for (int i=0;i<n;i++) a[i]*=b[i];
    FFT(a,-1);
    for (int i=0;i<n;++i) ans[i]=(int)(a[i].real()/n+0.5);
    for (int i=0;i<n;++i) if (ans[i]>9) ans[i+1]+=ans[i]/10,ans[i]%=10;
    while (!ans[n]) n--;
    for (int i=n;i>=0;i--) printf("%d",ans[i]);cout<<endl;
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值