更新一下多项式板子

#include<bits/stdc++.h>
#define MOD 998244353
#define ll long long
using namespace std;
inline int add(int a,int b)
{a+=b;return a>=MOD?a-MOD:a;}
inline int sub(int a,int b)
{a-=b;return a<0?a+MOD:a;}
inline int mul(int a,int b)
{return 1ll*a*b%MOD;}
int ksm(int a,int b)
{
	int ans=1;
	for(;b;b>>=1,a=mul(a,a))
		if(b&1)ans=mul(ans,a);
	return ans;
}
const int P=1<<18;
int w[P];
void NTT(int a[],int n,int ty)
{
	for(int i=0,j=0;i<n;i++){
		if(i<j)swap(a[i],a[j]);
		for(int l=n>>1;(j^=l)<l;l>>=1);
	}
	w[0]=1;
	for(int m=1;m<n;m<<=1){
		int owo=ksm(3,(MOD-1)/(m<<1)*ty+MOD-1);
		for(int i=1;i<m;i++)
			w[i]=mul(w[i-1],owo);
		for(int i=0;i<n;i+=(m<<1))
			for(int j=0;j<m;j++){
				int p=a[i+j],q=mul(a[i+j+m],w[j]);
				a[i+j]=add(p,q),a[i+j+m]=sub(p,q);
			}
	}
	if(ty<0){
		int inv=ksm(n,MOD-2);
		for(int i=0;i<n;i++)
			a[i]=mul(a[i],inv);
	}
}
int h1[P];
void Poly_Inv(int a[],int iv[],int n)
{
	if(n==1){
		iv[0]=ksm(a[0],MOD-2);
		return;
	}
	Poly_Inv(a,iv,n>>1);
	fill(iv+(n>>1),iv+(n<<1),0);
	for(int i=0;i<n;i++)
		h1[i]=a[i];
	fill(h1+n,h1+(n<<1),0);
	NTT(iv,n<<1,1),NTT(h1,n<<1,1);
	for(int i=0;i<(n<<1);i++)
		iv[i]=mul(iv[i],sub(2,mul(iv[i],h1[i])));
	NTT(iv,n<<1,-1);
}
int ha[P],hb[P],ihb[P],hc[P],hd[P];
void Poly_Mod(int a[],int b[],int n,int m)
{
	int maxn=1;
	while(maxn<n-m+1)maxn<<=1;
	for(int i=0;i<=n-m;i++)
		hb[i]=i<=m?b[m-i]:0,ha[i]=a[n-i];
	fill(hb+n-m+1,hb+maxn,0);
	Poly_Inv(hb,ihb,maxn);
	fill(ihb+maxn,ihb+(maxn<<1),0);
	fill(ha+n-m+1,ha+(maxn<<1),0);
	NTT(ha,maxn<<1,1),NTT(ihb,maxn<<1,1);
	for(int i=0;i<(maxn<<1);i++)
		hc[i]=mul(ha[i],ihb[i]);
	NTT(hc,maxn<<1,-1);
	reverse(hc,hc+n-m+1);
	for(maxn=1;maxn<=n;maxn<<=1);
	for(int i=0;i<=m;i++)
		hb[i]=b[i];
	fill(hb+m+1,hb+maxn,0);
	fill(hc+n-m+1,hc+maxn,0);
	NTT(hb,maxn,1),NTT(hc,maxn,1);
	for(int i=0;i<maxn;i++)
		hd[i]=mul(hb[i],hc[i]);
	NTT(hd,maxn,-1);
	for(int i=0;i<=n;i++)
		a[i]=sub(a[i],hd[i]);
}
int inv[P];
int hl[P];
void Poly_Ln(int a[],int l[],int n)
{
	Poly_Inv(a,l,n);
	fill(l+n,l+(n<<1),0);
	for(int i=1;i<n;i++)
		hl[i-1]=mul(a[i],i);
	fill(hl+n-1,hl+(n<<1),0);
	NTT(l,n<<1,1),NTT(hl,n<<1,1);
	for(int i=0;i<(n<<1);i++)
		l[i]=mul(l[i],hl[i]);
	NTT(l,n<<1,-1);
	for(int i=n-1;i;--i)
		l[i]=mul(l[i-1],inv[i]);
	l[0]=0;
}
int he[P];
void Poly_Exp(int a[],int e[],int n)
{
	if(n==1){
	//	e[0]=1;
		return;
	}
	Poly_Exp(a,e,n>>1);
	fill(e+(n>>1),e+(n<<1),0);
	Poly_Ln(e,he,n);
	fill(he+n,he+(n<<1),0);
	for(int i=0;i<n;i++)
		he[i]=sub(a[i],he[i]);
	he[0]=add(he[0],1);
	NTT(e,n<<1,1),NTT(he,n<<1,1);
	for(int i=0;i<(n<<1);i++)
		e[i]=mul(e[i],he[i]);
	NTT(e,n<<1,-1);
}
int hk[P];
void Poly_Ksm(int a[],int k[],int n,int kk)
{
	int va=ksm(a[0],kk);
	Poly_Ln(a,hk,n);
	for(int i=0;i<n;i++)
		hk[i]=mul(hk[i],kk);
	k[0]=va;
	Poly_Exp(hk,k,n);
}
int main()
{
	return 0;
}
posted @ 2020-07-21 20:01  蒟蒻小果冻  阅读(73)  评论(0编辑  收藏  举报