FFT各种模板

丑陋敬请谅解:

求两列数的卷积:

递归版:

#include <stdio.h>

#include <algorithm>

#include <math.h>

using namespace std;

const double pi=acos(-);

int p[],q[];

struct complex

{

    double x,y;

    complex(double xx=,double yy=)

    {

        x=xx;

        y=yy;

    }

}a[],b[];

complex operator +(complex a,complex b){return complex(a.x+b.x,a.y+b.y);}

complex operator -(complex a,complex b){return complex(a.x-b.x,a.y-b.y);}

complex operator *(complex a,complex b){return complex(a.x*b.x-a.y*b.y,a.x*b.y+a.y*b.x);}

void DFT(int limit,complex *a)

{

    if(limit==)

        return;

    int mid=limit>>;

    complex a1[mid],a2[mid];

    for(int i=;i<=limit;i+=)

    {

        a1[i>>]=a[i];

        a2[i>>]=a[i+];

    }

    DFT(mid,a1);

    DFT(mid,a2);

    complex wn=complex(cos(2.0*pi/limit),sin(2.0*pi/limit));

    complex w=complex(,);

    for(int i=;i<mid;i++,w=w*wn)

    {

        a[i]=a1[i]+w*a2[i];

        a[i+mid]=a1[i]-w*a2[i];

    }

}

int pow(int x,int y)

{

    if(y==)

        return ;

    int t=pow(x,y/);

    if(y%==)

        return t*t;

    return t*t*x;

}

void IDFT(int limit,complex *a)

{

    if(limit==)

        return;

    int mid=limit>>;

    complex a1[mid],a2[mid];

    for(int i=;i<=limit;i+=)

    {

        a1[i>>]=a[i];

        a2[i>>]=a[i+];

    }

    IDFT(mid,a1);

    IDFT(mid,a2);

    complex wn=complex(cos(2.0*pi/limit),-sin(2.0*pi/limit));

    complex w=complex(,);

    for(int i=;i<mid;i++,w=w*wn)

    {

        a[i]=a1[i]+w*a2[i];

        a[i+mid]=a1[i]-w*a2[i];

    }

}

int main()

{

    int k;

    scanf("%d",&k);

    int n=pow(,k);

    for(int i=;i<=n-;i++)

        scanf("%d",&p[i]);

    for(int i=;i<=n-;i++)

        scanf("%d",&q[i]);

    for(int i=;i<=n-;i++)

        a[i]=p[i];

    for(int i=;i<=n-;i++)

        b[i]=q[i];

    int limit=*n;

    DFT(limit,a);

    DFT(limit,b);

    for(int i=;i<=limit;i++)

        a[i]=a[i]*b[i];

    IDFT(limit,a);

    for(int i=;i<=n-;i++)

        printf("%d\n",(int)(a[i].x/limit+0.5));

    printf("%d",(int)(a[n-].x/limit+0.5));

}

非递归版+蝶形算法优化:

#include<stdio.h>

#include <algorithm>

#include<math.h>

using namespace std;

const int MAXN=1e7+;

const double Pi=acos(-1.0);

int p[MAXN],q[MAXN];

struct complex

{

        double x,y;

        complex (double xx=,double yy=){x=xx,y=yy;}

}a[MAXN],b[MAXN];

complex operator + (complex a,complex b){ return complex(a.x+b.x , a.y+b.y);}

complex operator - (complex a,complex b){ return complex(a.x-b.x , a.y-b.y);}

complex operator * (complex a,complex b){ return complex(a.x*b.x-a.y*b.y , a.x*b.y+a.y*b.x);}

int N,M;

int l,r[MAXN];

int limit=;

void DFT(complex *A)

{

        for(int i=;i<limit;i++)

                if(i<r[i]) swap(A[i],A[r[i]]);

        for(int mid=;mid<limit;mid<<=)

        {

                complex Wn( cos(Pi/mid) , sin(Pi/mid) );

                for(int R=mid<<,j=;j<limit;j+=R)

                {

                        complex w(,);

                        for(int k=;k<mid;k++,w=w*Wn)

                        {

                                 complex x=A[j+k],y=w*A[j+mid+k];

                                A[j+k]=x+y;

                                A[j+mid+k]=x-y;

                        }

                }

        }

}

void IDFT(complex *A)

{

        for(int i=;i<limit;i++)

                if(i<r[i]) swap(A[i],A[r[i]]);

        for(int mid=;mid<limit;mid<<=)

        {

                complex Wn( cos(Pi/mid) , -1.0*sin(Pi/mid) );

                for(int R=mid<<,j=;j<limit;j+=R)

                {

                        complex w(,);

                        for(int k=;k<mid;k++,w=w*Wn)

                        {

                                 complex x=A[j+k],y=w*A[j+mid+k];

                                A[j+k]=x+y;

                                A[j+mid+k]=x-y;

                        }

                }

        }

}

int main()

{

        scanf("%d%d",&N,&M);

        for(int i=;i<=N;i++) scanf("%d",&p[i]);

        for(int i=;i<=M;i++) scanf("%d",&q[i]);

          for(int i=;i<=N;i++) a[i]=p[i];

        for(int i=;i<=M;i++)  b[i]=q[i];

        while(limit<=N+M) limit<<=,l++;

        for(int i=;i<limit;i++)

                r[i]= ( r[i>>]>> )| ( (i&)<<(l-) ) ;

        DFT(a);

        DFT(b);

        for(int i=;i<=limit;i++) a[i]=a[i]*b[i];

        IDFT(a);

        for(int i=;i<=N+M;i++)

                printf("%d ",(int)(a[i].x/limit+0.5));

        return ;

}

FFT版高精度乘法:

#include<stdio.h>

#include <algorithm>

#include<math.h>

#include <string.h>

using namespace std;

const double Pi=acos(-1.0);

char s1[],s2[];

int sum[];

struct complex

{

    double x,y;

    complex (double xx=,double yy=){x=xx,y=yy;}

}a[],b[];

complex operator + (complex a,complex b){ return complex(a.x+b.x , a.y+b.y);}

complex operator - (complex a,complex b){ return complex(a.x-b.x , a.y-b.y);}

complex operator * (complex a,complex b){ return complex(a.x*b.x-a.y*b.y , a.x*b.y+a.y*b.x);}

int N,M,L;

int len=;

int l,r[];

int limit=;

void  fft(complex *A,int limit,int type)

{

    for(int i=;i<limit;i++)

        if(i<r[i]) swap(A[i],A[r[i]]);

    for(int mid=;mid<limit;mid<<=)

    {

        complex Wn( cos(Pi/mid) , type*sin(Pi/mid) );

        for(int R=mid<<,j=;j<limit;j+=R)

        {

            complex w(,);

            for(int k=;k<mid;k++,w=w*Wn)

            {

                 complex x=A[j+k],y=w*A[j+mid+k];

                A[j+k]=x+y;

                A[j+mid+k]=x-y;

            }

        }

    }

    if(type==-)for(int i=;i<limit;i++)A[i].x/=limit;

}

int main()

{

    int n;

    scanf("%d",&n);

    scanf("%s",s1);

    scanf("%s",s2);

    int len1=strlen(s1);

    int len2=strlen(s2);

        int lenx=len1+len2;

        for(len=;len<lenx;len<<=)

            L++;

        for(int i=;i<len;i++)

            r[i]=((r[i>>])>>)|((i&)<<(L-));

        for(int i=;i<len1;i++)

            a[i]=complex(s1[len1-i-]-'',);

        for(int i=len1;i<len;i++)

            a[i]=complex(,);

        for(int i=;i<len2;i++)

            b[i]=complex(s2[len2-i-]-'',);

        for(int i=len2;i<len;i++)

            b[i]=complex(,);

         fft(a,len,);fft(b,len,);

        for(int i=;i<len;i++)

            a[i]=a[i]*b[i];

        fft(a,len,-);

        for(int i=;i<len;i++)

            sum[i]=int(a[i].x+0.5);

        for(int i=;i<len;i++)

        {

            sum[i+]+=sum[i]/;

            sum[i]%=;

        }

    len=len1+len2-;

        while(sum[len]<= && len>)

            len--;

        for(int i=len;i>=;i--)

            printf("%c",sum[i]+'');

    return ;

}

FFT各种模板的更多相关文章

FFT快速傅里叶模板
FFT快速傅里叶模板…… /* use way: assign : h(x) = f(x) * g(x) f(x):len1 g(x):len2 1. len = 1; while(len < ...
多项式FFT相关模板
自己码了一个模板...有点辛苦...常数十分大,小心使用 #include <iostream> #include <stdio.h> #include <math.h& ...
多项式FFT/NTT模板(含乘法/逆元/log/exp/求导/积分/快速幂)
自己整理出来的模板存在的问题: 1.多项式求逆常数过大(尤其是浮点数FFT) 2.log只支持f[0]=1的情况,exp只支持f[0]=0的情况有待进一步修改和完善 FFT: #include&l ...
hdu 1402 FFT（模板）
A * B Problem Plus Time Limit: 2000/1000 MS (Java/Others) Memory Limit: 65536/32768 K (Java/Other ...
FFT/NTT模板既 HDU1402 A * B Problem Plus
@(学习笔记)[FFT, NTT] Problem Description Calculate A * B. Input Each line will contain two integers A a ...
FFT NTT 模板
NTT: #include<cstdio> #include<cstring> #include<algorithm> using namespace std; # ...
多项式乘法（FFT）模板 && 快速数论变换（NTT）
具体步骤: 1.补0:在两个多项式最前面补0,得到两个 $2n$ 次多项式,设系数向量分别为 $v_1$ 和 $v_2$. 2.求值:用FFT计算 $f_1 = DFT(v_1)$ 和 $f_2=DF ...
分治FFT/NTT 模板
题目要我们求$f[i]=\sum\limits_{j=1}^{i}f[i-j]g[j]\;mod\;998244353$ 直接上$NTT$肯定是不行的,我们不能利用尚未求得的项卷积所以要用$CDQ$ ...
UOJ#34 FFT模板题
写完上一道题才意识到自己没有在博客里丢过FFT的模板-- 这道题就是裸的多项式乘法,可以FFT,可以NTT,也可以用Karasuba(好像有人这么写没有T),也可以各种其他分治乘法乱搞-- 所以我就直 ...

随机推荐

JMeter配置JDBC测试SQL Server/MySQL/ORACLE
一.配置SQL Server 1.下载sql驱动,将sqljdbc4.jar放到JMeter安装目录/lib下. 2.启动JMeter,右键添加->配置文件->JDBC Connectio ...
Random Variables
可测空间(Measurable Space)和测度空间(Measure Space) 集合X,X上的一个σ-algebra A,则(X,A)被称为可测空间(measurable space) 再在A上 ...
Wepy框架和mpVue框架的比较及使用mpVue框架需要注意的点
Wepy框架它是一款类Vue框架,在代码风格上借鉴了Vue,本身和Vue没有任何关系. mpVue框架它是从整个Vue的核心代码上经过二次开发而形成的一个框架,相当于是给Vue本身赋能,增加了开发 ...
nrm 源管理器
什么是nrm nrm 是一个 npm 源管理器,允许你快速地在 npm 源间切换. 安装nrm 在命令行执行命令,npm install -g nrm,全局安装nrm. 使用执行命令nrm ls查看 ...
吴裕雄 Bootstrap 前端框架开发——Bootstrap 按钮：按钮被点击
<!DOCTYPE html> <html> <head> <meta charset="utf-8"> <title> ...
netsh命令获取wifi历史连接密码
首先[win+r]快捷键打开运行,输入cmd.或点击左下角win-运行-cmd 1.netsh wlan show profiles //列出所有的ap名称 2.netsh wlan show pro ...
JS 自动返回每个月的天数
);//M代表月份,可以自己手动设置或者选择 date.setDate() var num = date.getDate(); console.log(num) //输出每个月的天数 var time ...
VUE框架下安装自带http协议
在控制台CMD 中输入 npm install vue-resource --save-dev
u盘装完centos系统恢复
1.使用windows的cmd窗口,执行diskpart命令 2.执行 list disk命令,查看u盘 3.执行 select disk 2,选中u盘,注意,这里的2是我自己的显示,千万不要选错 4 ...
ActivePerl 安装
下载 https://www.activestate.com/products/activeperl/downloads/ 链接:https://pan.baidu.com/s/1IXPdYFd5bD ...

FFT各种模板

FFT各种模板的更多相关文章

随机推荐

热门专题