NTT(快速数论变换)

用途: 计算整系数多项式卷积。

前置芝士

FFT

原根

作用

我们发现，很多多项式系数为整数，而 $FFT$ 引入了毒瘤的虚数，会出现精度误差且运算速度较慢，这时候就需要一个基于整数的的算法来计算多项式的卷积。

做法

我们先来考虑为什么 $FFT$ 要用到虚数( $n$ 次单位根作为点表示法的横坐标)。

因为它能优化运算次数(废话)。那是什么让他可以优化运算次数呢?

FFT板子

#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define LL long long 
#define LB long double
#define N 3000006
using namespace std;

const int G=3;
const int IG=332748118;
const int mod=998244353;

int a[N],b[N];
int n,m,bit,tot,rev[N];

inline int qr()
{
	char a=0;int w=1,x=0;
	while(a<'0'||a>'9'){if(a=='-')w=-1;a=getchar();}
	while(a<='9'&&a>='0'){x=(x<<3)+(x<<1)+(a^48);a=getchar();}
	return x*w;
}

inline int poww(int a,int x)
{
	int ans=1;
	while(x)
	{
		if(x&1)
			ans=((LL)ans*a)%mod;
		a=((LL)a*a)%mod;
		x>>=1;
	}
	return ans;
}

int exgcd(int a,int b,int &x,int &y)
{
	if(!b)
	{
		x=1,y=0;
		return a;
	}
	int d=exgcd(b,a%b,y,x);
	y-=(a/b)*x;
	return d;
}

inline int inv(int p)
{
	int x,y;
	exgcd(p,mod,x,y);
	return (x%mod+mod)%mod;
}

inline int add(int a,int b)
{
	return a+b>mod?a+b-mod:a+b;
}

inline int sub(int a,int b)
{
	return a-b<0?a-b+mod:a-b;
}

inline void NTT(int a[],int opt)
{
	for(register int i=0;i<tot;i++)
		if(i<rev[i])
			swap(a[i],a[rev[i]]);
	for(register int mid=1;mid<tot;mid<<=1)
	{
		int w1=poww(opt==1?G:IG,(mod-1)/(mid<<1));
		for(register int i=0;i<tot;i+=(mid<<1))
		{
			int wk=1;
			for(register int j=0;j<mid;j++)
			{
				int x=a[i+j];
				int y=(LL)wk*a[i+j+mid]%mod;
				a[i+j]=add(x,y);
				a[i+j+mid]=sub(x,y);
				wk=(LL)wk*w1%mod;
			}
		}
	}
}

int main()
{
	n=qr();
	m=qr();
	for(register int i=0;i<=n;i++)
		a[i]=qr();
	for(register int i=0;i<=m;i++)
		b[i]=qr();
	while((1<<bit)<n+m+1)
		bit++;
	tot=1<<bit;
	for(register int i=0;i<tot;i++)
		rev[i]=(rev[i>>1]>>1)|((i&1)<<(bit-1));
	NTT(a,1);
	NTT(b,1);
	for(register int i=0;i<tot;i++)
		a[i]=(LL)a[i]*b[i]%mod;
	NTT(a,-1);
	int inv_=inv(tot);
	for(register int i=0;i<n+m+1;i++)
		printf("%lld ",((LL)a[i]*inv_)%mod);
	printf("\n");
	return 0;
}