hdu1402 A*B 快速傅里叶变换(FFT)

      两个不超过50000位的数,求乘积。高精度乘法,但是写暴力高精的话复杂度是O(n^2),不压位估计是要T掉的=,这里介绍一种新的方法,可以在O(nlogn)的复杂度内求出答案。

先来说一下我对fft的理解吧,fft其实就是一个求多项式乘法的快速算法,两个n阶多项式相乘,传统的方法是循环相乘再累加,复杂度是O(n^2),而用fft去实现的话,复杂度可以降到O(nlogn)。对于n阶多项式,我们可以用两种方法去表示,一种是常用的稀疏表示法,也就是a0+a1*x^1+a2*x^2+...+an*x^n,另一种是点集表示法,在平面上取x坐标的集合X,对于每个x[i]求出y[i]=A(x),其中A()为一个多项式,这样可以用这个点集来描述这个多项式(可以看做是集合表示吧)。对于用系数表示两个多项式X,Y,求Z=X*Y的话,可以通过一下四步实现:

1:使次数界增加一倍:把X,Y的系数项扩大一倍,扩充成2n阶的多项式。

2: 求值:把X和Y分别转化成用的点集表示法表示。

3:点乘:Z[i]=X[i]*Y[i],乘法可以看做复数乘法。

4:插值:把Z由点集表示法转化成稀疏表示法。

其中过程2即离散傅里叶变化(DFT),过程4为逆离散傅里叶变换(IDFT),而如果有选择性的选取某些点来构成点集表示发的集合(即选取复平面上的“单位复根”),就可以将DFT优化成O(nlogn)的FFT从而降低整体的时间复杂度,最终过程2,4的时间复杂度为O(nlogn),过程1,3的复杂度为O(n),整体的时间复杂度为O(nlogn).FFT具体的具体实现可以参考算导上的讲解,当然看不明白的话就只好记模板了= -...


然后来看这一题,两个数相乘的话,可以看做x[i]=10^i的两个多项式的乘法,那么可以直接用FFT来实现,要注意的一点就是求出乘积的系数后,转化成整数时注意四舍五入,然后注意处理进位的情况。

#include 
#include 
#include 
#include 
#include 
using namespace std;
typedef long long ll;
const double PI = acos(-1.0);
struct comp
{
    double r,i;
    comp(double rt=0,double it=0)
    {
        r=rt;
        i=it;
    }
    comp operator +(const comp& b)
    {
        return comp(r+b.r,i+b.i);
    }
    comp operator -(const comp &b)
    {
        return comp(r-b.r,i-b.i);
    }
    comp operator *(const comp &b)
    {
        return comp(r*b.r-i*b.i,r*b.i+i*b.r);
    }
};

void change(comp y[],int len)//二进制转置--雷德算法
{
    int i,j,k;
    for(i = 1, j = len/2;i < len-1;i++)
    {
        if(i < j)swap(y[i],y[j]);
        k = len/2;
        while( j >= k)
        {
            j -= k;
            k /= 2;
        }
        if(j < k)j += k;
    }
}

void fft(comp y[],int len,int on)
/* on=1 DFT 把一个多项式的系数向量转化为点集表示;
on=-1,IDFT 把一个点集转化成多项式的系数向量*/
{
    change(y,len);
    for(int h = 2;h <= len;h <<= 1)
    {
        comp wn(cos(-on*2*PI/h),sin(-on*2*PI/h));
        for(int j = 0;j < len;j += h)
        {
            comp w(1,0);
            for(int k = j;k < j+h/2;k++)
            {
                comp u = y[k];
                comp t = w*y[k+h/2];
                y[k] = u+t;
                y[k+h/2] = u-t;
                w = w*wn;
            }
        }
    }
    if(on == -1)
        for(int i = 0;i < len;i++)
            y[i].r /= len;
}

void conv(comp f[],int len)//求f的卷积
{
    fft(f,len,1);
    for (int i=0; i9)
        {
            ans[i+1]+=ans[i]/10;
            ans[i]%=10;
        }

        bool ok=false;
        for (int i=len-1; i>=0; i--)
        {
            if (ok) printf("%d",ans[i]);
            else if (ans[i])
            {
                ok=true;
                printf("%d",ans[i]);
            }
        }
        if (!ok) puts("0");
        else puts("");


    }
}


你可能感兴趣的:(math)