学习链接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;
}