zoukankan      html  css  js  c++  java
  • FFT(快速傅里叶变换)

    学习了FFT用来求多项式的乘法,看了算导上的介绍,上面讲的非常明白,概括一下FFT的原理就是,我们在计算多项式的乘法时,如果暴力模拟的话是n^2 复杂度的,就像小学学的竖式乘法一样,比如一个n位数乘上一个n位数,我们需要用竖式乘法计算要列n行,每一行有n个元素,然后相邻两行错开一位(很显然,竖式乘法谁都会),如果我们换一种形式呢?有一种表达是叫做点值表达,就好像我们上了初中学二次函数,如果已知函数图像上的三个不同的点坐标,我们可以写出函数的表达式,那么就是说利用函数图象上的点我们也可以确定一个函数(多项式),这样的好处在于点值表达式更加容易进行乘法运算,举个例子:

    我们现在有两个函数: y1=2*x*x+5*x+1  y2=x*x+2*x+1  我们要求y1*y2的表达式。如果利用系数表达,我们利用竖式乘法可以手算出答案,复杂度n^2.

    由于两个函数是二次的,乘起来是四次的,所以我们需要知道5个点的坐标。

    x 1 2 3 4 5
    y1 8 19 34 53 76
    y2 4 9 16 25 36
    y 32 171 544 1325 2736

    我们现在已知了y的五个坐标,那么肯定可以求出y的表达式,那么多项式乘法也就做完了。

    可以看出对于点值表达,它的乘法操作是O(N)的,只需要把对应的y值乘起来即可。

    那么FFT算法就诞生了,他就是利用点值表达的优点,把系数表达转换成点值表达,然后再转化回去。

    从系数表达到点值表达的过程交做求值,利用秦九韶公式我们可以O(N)的算出一个点对应的y值。

    从点值表达到系数表达的过程叫做差值,著名的拉格朗日插值法复杂度是O(N^2)的,这对于我们多项式乘法没有太大帮助。

    因为差值对于我们取哪些点并没有要求,所以我们选取一些特殊点,利用一些特殊性质可以优化插值的过程。

    利用了复数根的知识,大概原理就是,我们可以分治的求一个系数向量,每一次递归求一半,然后可以O(N)的合并,在算导的第三十章都有讲到,推荐大家去看算导,里面讲的非常详细易懂。由于我太菜了,讲不清其中的奥秘。

    所以只能在这里给出程序代码了,我没有进行递归的求解,进行了迭代操作,看一下算导上30-4的图就明白r数组的含义了,其实就是计算的顺序。 —— by VANE

    #include<bits/stdc++.h>
    #define N 3000005 
    #define pi acos(-1)
    using namespace std;
    typedef complex<double> E;
    int n,m,l,r[N];
    E a[N],b[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 p=i<<1,j=0;j<n;j+=p)
            {
                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;
                }
            }
        }
    }
    inline int read()
    {
        char ch=getchar();int f=1,x=0;
        while(ch<'0'||ch>'9') {if(ch=='-') f=-f;ch=getchar();}
        while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
        return x*f;
    }
    int main()
    {
        n=read();m=read();
        for(int i=0;i<=n;i++)a[i]=read();
        for(int i=0;i<=m;i++)b[i]=read();
        m+=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]=a[i]*b[i];
        FFT(a,-1);
        for(int i=0;i<=m;i++)printf("%d ",(int)(a[i].real()/n+0.5));
    }

     附上一个高精度乘法,其实就是FFT的简单变形。

    // luogu-judger-enable-o2
    #include<bits/stdc++.h>
    #define N 300005 
    #define pi acos(-1)
    using namespace std;
    typedef complex<double> E;
    int n,m,l,r[N],ans[N];
    E a[N],b[N];
    char s[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 p=i<<1,j=0;j<n;j+=p)
            {
                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;
                }
            }
        }
    }
    inline int read()
    {
        char ch=getchar();int f=1,x=0;
        while(ch<'0'||ch>'9') {if(ch=='-') f=-f;ch=getchar();}
        while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
        return x*f;
    }
    int main(){
        n=read();n--;m=n;
        scanf("%s",s);
        for(int i=0;i<=n;i++)a[i]=s[n-i]-'0';
        scanf("%s",s);
        for(int i=0;i<=m;i++)b[i]=s[m-i]-'0';
        m+=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]=a[i]*b[i];
        FFT(a,-1);
        for(int i=0;i<=m;i++)ans[i]=(int)(a[i].real()/n+0.5);
        for(int i=0;i<=m;++i)
        ans[i+1]+=ans[i]/10,ans[i]%=10;
        if(ans[m+1]) m++;
        while(ans[m]==0&&m) m--;
        for(int i=m;i>=0;--i) printf("%d",ans[i]);
    }
  • 相关阅读:
    day09总结
    day09作业
    删除循环中选中的id
    数组套数组 获取数据
    vue for each循环出来数组的某一项 再重新合并成一个新数组
    全是没见过的
    vue element 分页
    解决iphonex屏幕过长背景图片或者放在元素里面的图片不能铺满的问题
    实在自动现在APK,微信跳浏览器下载
    js判断是否在微信浏览器中打开
  • 原文地址:https://www.cnblogs.com/nbwzyzngyl/p/8329786.html
Copyright © 2011-2022 走看看