正題

題目連結:https://www.luogu.com.cn/problem/P5333

題目大意

給出\(n\)棵樹，第\(i\)棵樹有\(k_i\)個點，每棵樹上的每個點和其它樹上的所有點都有連邊。

求這棵樹有多少條哈密頓迴路。

答案對\(998244353\)取模。

\(\sum_{i=1}^nk_i\leq 5000\)

解題思路

我們把每棵樹分成若干條路徑，那麼這個哈密頓迴路肯定是這些路徑拼起來，並且相鄰的路徑不能來自於同一棵樹。

那麼對於每棵樹我們先算出一個\(g_i\)表示將這棵樹分成\(i\)條路徑的方案數。

至於相鄰的路徑不能來自於同一棵樹的限制我們考慮容斥，我們硬將一棵樹上\(i\)

然後揹包出總共\(i\)條路徑的方案，像可重排一樣在前面一棵樹上\(j\)條路徑的就乘上\(\frac1{j!}\)，最後乘上\(i!\)就好了。

然後為了防止算重我們固定從\(1\)出發，那麼處理第一棵樹的時候我們就預設有一個放最前面，然後再減去頭尾都來自第\(1\)棵樹的方案就好了。

時間複雜度：\(O((\sum k_i)^2)\)

code

#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
const ll N=5100,P=998244353;
struct node{
	ll to,next;
}a[N<<1];
ll n,m,sum,tot,ls[N],g[N],h[N];
ll siz[N],f[N][N][3],tmp[N][3];
ll fac[N],inv[N];
void addl(ll x,ll y){
	a[++tot].to=y;
	a[tot].next=ls[x];
	ls[x]=tot;return;
}
void dfs(ll x,ll fa){
	siz[x]=1;f[x][0][2]=1;
	for(ll i=ls[x];i;i=a[i].next){
		ll y=a[i].to;
		if(y==fa)continue;
		dfs(y,x);
		for(ll j=0;j<=siz[x];j++)
			for(ll k=0;k<=siz[y];k++){
				(tmp[j+k][2]+=f[x][j][2]*f[y][k][0]%P)%=P;
				(tmp[j+k][1]+=f[x][j][2]*f[y][k][1]%P)%=P;
				(tmp[j+k][1]+=f[x][j][1]*f[y][k][0]%P)%=P;
				(tmp[j+k][0]+=f[x][j][0]*f[y][k][0]%P)%=P;
				(tmp[j+k+1][0]+=f[x][j][1]*f[y][k][1]*2ll%P)%=P;
			}
		siz[x]+=siz[y];
		for(ll j=0;j<=siz[x];j++)
			for(ll k=0;k<3;k++)
				f[x][j][k]=tmp[j][k],tmp[j][k]=0;
	}
	for(ll i=0;i<=siz[x];i++){
		(f[x][i+1][0]+=f[x][i][1]*2ll)%=P;
		(f[x][i+1][0]+=f[x][i][2])%=P;
		(f[x][i][1]+=f[x][i][2])%=P;
	}
	return;
}
ll C(ll n,ll m)
{return fac[n]*inv[m]%P*inv[n-m]%P;}
signed main()
{
	fac[0]=inv[0]=inv[1]=h[0]=1;
	for(ll i=2;i<N;i++)inv[i]=P-inv[P%i]*(P/i)%P;
	for(ll i=1;i<N;i++)fac[i]=fac[i-1]*i%P,inv[i]=inv[i-1]*inv[i]%P;
	scanf("%lld",&m);
	for(ll p=1;p<=m;p++){
		scanf("%lld",&n);
		for(ll i=1;i<=n;i++)ls[i]=0;tot=0;
		for(ll i=1,x,y;i<n;i++){
			scanf("%lld%lld",&x,&y);
			addl(x,y);addl(y,x);
		}
		dfs(1,0);sum+=n;
		for(ll i=0;i<=n;i++)g[i]=0;
		if(p!=1){
			for(ll i=1;i<=n;i++)
				for(ll j=i;j<=n;j++)
					(g[i]+=f[1][j][0]*fac[j]%P*C(j-1,i-1)%P*(((j-i)&1)?(P-1):1)%P)%=P;
		}
		else{
			for(ll i=1;i<=n;i++)
				for(ll j=i;j<=n;j++)
					(g[i-1]+=f[1][j][0]*fac[j-1]%P*C(j-1,i-1)%P*(((j-i)&1)?(P-1):1)%P)%=P;
			for(ll i=2;i<=n;i++)
				for(ll j=i;j<=n;j++)
					(g[i-2]+=f[1][j][0]*fac[j-1]%P*C(j-1,i-1)%P*(((j-i)&1)?1:(P-1))%P)%=P;
		}
		for(ll i=1;i<=n;i++)g[i]=g[i]*inv[i]%P;
		for(ll j=sum;j>=0;j--){
			ll tmp=0;
			for(ll i=0;i<=min(n,j);i++)
				(tmp+=h[j-i]*g[i]%P)%=P;
			h[j]=tmp;
		}
		for(ll i=1;i<=n;i++)
			for(ll j=0;j<=siz[i];j++)
				for(ll k=0;k<3;k++)f[i][j][k]=0;
	}
	ll ans=0;
	for(ll i=0;i<=sum;i++)
		(ans+=h[i]*fac[i]%P)%=P;
	printf("%lld\n",ans);
	return 0;
}