FFT（快速傅里叶变换）模板

学习链接1：点击进入

学习链接2：点击进入

主要用处

O(nlogn)计算多项式乘法。
DFT（离散傅里叶变换）：多项式转化成“离散的点”，系数表示法变成点值表示法.
IDFT（离散傅里叶反变换）把“离散的点”还原回多项式，点值表示法变成系数表示法。
… …

代码（模板）

#include<iostream>
#include<string>
#include<map>
#include<set>
//#include<unordered_map>
#include<queue>
#include<cstdio>
#include<vector>
#include<cstring>
#include<stack>
#include<algorithm>
#include<iomanip>
#include<cmath>
#include<fstream>
#define X first
#define Y second
#define best 131 
#define INF 0x3f3f3f3f3f3f3f3f
#define pii pair<int,int>
#define lowbit(x) x & -x
#define inf 0x3f3f3f3f
//#define int long long
//#define double long double
//#define rep(i,x,y) for(register int i = x; i <= y;++i)
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const double pai=acos(-1.0);
const int maxn=3e6+10;
const int mod=998244353;
const double eps=1e-9;
const int N=5e3+10;
char s1[maxn],s2[maxn];
int n,m,ans[maxn];
int lim=1,len,rev[maxn]; 
struct node
{
    double x,y;
    node(double xx=0,double yy=0){x=xx,y=yy;}
    node operator * (node Q){return node(x*Q.x-y*Q.y,x*Q.y+y*Q.x);}
	node operator + (node Q){return node(x+Q.x,y+Q.y);}
	node operator - (node Q){return node(x-Q.x,y-Q.y);}
}a[maxn],b[maxn];

inline int read()
{
    int k = 0, f = 1 ;
    char c = getchar() ;
    while(!isdigit(c)){if(c == '-') f = -1 ;c = getchar() ;}
    while(isdigit(c)) k = (k << 1) + (k << 3) + c - 48 ,c = getchar() ;
    return k * f ;
}
//A表示要操作的系数 ,若flag为1，则表示DFT，为-1则为IDFT(需要求倒数） 
inline void fft(node *A,double flag)
{
    for(int i=0;i<lim;i++)
        if(i<rev[i]) 
			swap(A[i],A[rev[i]]);//i小于rev[i]时才交换，防止同一个元素交换两次，回到它原来的位置。 
    for(int i=1;i<lim;i<<=1)//枚举区间长度，从小区间到大区间依次合并，i是准备合并序列的长度的二分之一
	{
        node wn(cos(pai/i),flag*sin(pai/i)) ;//求单位根w_n^1 
        for(int j=0;j<lim;j+=(i<<1))//两段区间两段区间的合并，用于合并 
		{
            node w(1,0);
            for(int k=0;k<i;k++,w=w*wn) //枚举j所枚举的两个区间的值用于蝴蝶合并，求w_n^k 
			{
                node nx=A[j+k],ny=w*A[j+i+k];
                A[j+k]=nx+ny;//这两步是蝴蝶变换 
                A[j+i+k]=nx-ny ;
            }
        }
    }
}
int main()
{
    cin>>n>>m; 
	//读入的数的每一位看成多项式的一项，保存在复数的实部 
    for(int i=0;i<=n;i++) cin>>a[i].x;
    for(int i=0;i<=m;i++) cin>>b[i].x;
    //len表示转化成二进制的位数 lim为结果多项式的长度（暂时），化为2的幂次便于分治（二分） 
    while(lim<n+m+2) lim<<=1,len++;
    //初始化每个位置最终到达的位置 
    for(int i=0;i<=lim;i++) rev[i]=(rev[i>>1]>>1)|((i&1)<<(len-1)) ;
    fft(a,1);//DFT 把a的系数表示转化为点值表示 
	fft(b,1);//DFT 把b的系数表示转化为点值表示 
    for(int i=0;i<=lim;i++) a[i]=a[i]*b[i];//FFT 两个多项式的点值表示相乘 
    fft(a,-1);//IDFT 把这个点值表示转化为系数表示 
	for(int i=0;i<=n+m;i++) //保存答案的每一位(注意进位） 
        cout<<(int)(a[i].x/lim+0.5)<<' ';//取实数四舍五入，此时虚数部分应当为0或由于浮点误差接近0
    return 0;
}