bzoj2179 FFT快速傅立叶 （FFT）

bzoj2179 FFT快速傅立叶

原题地址：http://www.lydsy.com/JudgeOnline/problem.php?id=2179

题意：
给出两个n位10进制整数x和y，你需要计算x*y。

数据范围
n<=60000

题解：
FFT裸题。
感谢这篇FFT讲解得十分透彻的博文
注意，有精度差异，四舍五入加了0.5。

代码：

#include<cstdio>
#include<iostream>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
const double PI=acos(-1);
const int N=300005;
struct Virt
{
    double r,i;
    Virt(double r=0.0,double i=0.0) {this->r=r; this->i=i;}
    Virt operator+(const Virt &x) {return Virt(r+x.r,i+x.i);}
    Virt operator-(const Virt &x) {return Virt(r-x.r,i-x.i);}
    Virt operator*(const Virt &x) {return Virt(r*x.r-i*x.i,r*x.i+i*x.r);}
};
Virt conj(Virt x) {return Virt(x.r,-x.i);}
Virt a[N],b[N],omg[N],_omg[N];
char s[N];
int len,m,R[N],p,c[N];
void init()
{
    m=m*2-1; for(p=0,len=1;len<m;len=len<<1,p++);
    for(int i=1;i<len;i++) R[i]=(R[i>>1]>>1)|((i&1)<<(p-1));
    for(int k=0;k<len;k++) {omg[k]=Virt(cos(2*PI/len*k),sin(2*PI/len*k)); _omg[k]=conj(omg[k]);}
}
void FFT(Virt *f,int opt)
{
    Virt* w; if(opt==1) w=omg; else w=_omg;
    for(int i=0;i<len;i++) if(i<R[i]) swap(f[i],f[R[i]]);
    for(int i=2;i<=len;i<<=1)
    {
        int l=i>>1;
        for(int j=0;j<len;j+=i)
        {
            for(int k=0;k<l;k++)
            {
                Virt z=f[j+k+l]*w[len/i*k];
                f[j+k+l]=f[j+k]-z;
                f[j+k]=f[j+k]+z;
            }
        }
    }
    if(opt==-1) for(int i=0;i<len;i++) f[i].r=f[i].r/(double)len;
}
int main()
{
    scanf("%d",&m);
    scanf("%s",s); for(int i=0;i<m;i++) a[i].r=s[m-i-1]-'0';
    scanf("%s",s); for(int i=0;i<m;i++) b[i].r=s[m-i-1]-'0';
    init();
    FFT(a,1); FFT(b,1);
    for(int i=0;i<len;i++) a[i]=a[i]*b[i];
    FFT(a,-1);
    for(int i=0;i<len;i++) c[i]=(int)(a[i].r+0.5); // 精度差，四舍五入 
    for(int i=0;i<len;i++)  if(c[i]>=10) {c[i+1]+=c[i]/10; c[i]%=10; if(i==len-1)len++;}
    while(!c[len-1]) len--;
    for(int i=len-1;i>=0;i--) printf("%d",c[i]);
    return 0;
}