NTT和多项式各种基础运算模板

#include <bits/stdc++.h>
#define mod 998244353
#define N 400009
#define ll long long
using namespace std;
ll w[2][N],inv[N];
ll A[N],B[N],C[N],D[N],E[N],F[N];
ll W,G;
ll pow(ll a,ll b,ll c)
{
	ll ans=1;
	for (;b;a=a*a%c,b>>=1)
		if (b&1) ans=ans*a%c;
	return ans;
}
void dft(ll *a,ll n,bool v)
{
	for (ll i=0,j=0;i<n;i++)
	{
		if (i>j) swap(a[i],a[j]);
		for (ll l=n>>1;(j^=l)<l;l>>=1);
	}
	for (ll i=2;i<=n;i<<=1)
		for (ll j=0;j<n;j+=i)
			for (ll l=0;l<i>>1;l++)
			{
				ll t=a[j+l+(i>>1)]*w[v][(n/i)*l]%mod;
				a[j+l+(i>>1)]=(a[j+l]-t+mod)%mod;
				a[j+l]=(a[j+l]+t)%mod;
			}
	if (v)
		for (ll i=0,t=pow(n,mod-2,mod);i<n;i++) a[i]=a[i]*t%mod;
}
void pre(ll n)
{
	w[0][0]=w[0][n]=w[1][0]=w[1][n]=1;
	ll G=3;
	G=pow(G,(mod-1)/n,mod);
	for (ll i=1;i<n;i++) w[0][i]=w[0][i-1]*G%mod;
	for (ll i=1;i<n;i++) w[1][i]=w[0][n-i];
}
void Mul(ll *a,ll *b,ll n)
{
	pre(n<<=1);
	memset(E,0,sizeof(E));
	for (ll i=0;i<(n>>1);i++) E[i]=b[i];
	dft(a,n,0);
	dft(E,n,0);
	for (ll i=0;i<n;i++) a[i]=a[i]*E[i]%mod;
	dft(a,n,1);
	n>>=1;
	for (ll i=n;i<(n<<1);i++) a[i]=0;
}
void Inv(ll *a,ll *b,ll n)
{
	if (n==1)
	{
		b[0]=pow(a[0],mod-2,mod);
		return;
	}
	Inv(a,b,n>>1);
	for (ll i=0;i<n;i++) D[i]=a[i];
	for (ll i=n;i<(n<<1);i++) D[i]=0;
	pre(n<<=1);
	dft(D,n,0);
	dft(b,n,0);
	for (ll i=0;i<n;i++)
		b[i]=(b[i]*2%mod-D[i]*b[i]%mod*b[i]%mod+mod)%mod;
	dft(b,n,1);
	n>>=1;
	for (ll i=n;i<(n<<1);i++) b[i]=0;
}
void differentiate(ll *a,ll *b,ll n)
{
	b[Tn-1]=0;
	for (ll i=1;i<n;i++)
		b[i-1]=a[i]*i%mod;
}
void integrate(ll *a,ll *b,ll n)
{
	for (ll i=n-1;i;i--)
		b[i]=a[i-1]*inv[i]%mod;
	b[0]=0;
}
void ln(ll *a,ll *b,ll n)
{
	for (ll i=0;i<(n<<1);i++) A[i]=B[i]=0;
	Inv(a,A,n);
	differentiate(a,B,n);
	Mul(A,B,n);
	integrate(A,b,n);
}
void exp(ll *a,ll *b,ll n)
{
	if (n==1)
	{
		b[0]=1;
		return;
	}
	exp(a,b,n>>1);
	ln(b,C,n);
	for (ll i=0;i<n;i++) C[i]=a[i]-C[i];
	C[0]++;
	Mul(b,C,n);
}