YbtOJ NOIP2020 模拟赛 B 组 Day8 C.树上排列【dp】【组合计数】

---

题目：

传送门

---

题意：

给出一棵$n$个节点的树，对于每条边都有端点的大小关系，我们需要给这些点标号，满足编号是$n$的一个排列，问能有多少种方案数

---

分析：

---

代码：

#include<iostream>
#include<cstdio>
#include<string>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
#define LL long long 
#define mo 998244353
using namespace std;
inline LL read()
{
	LL s=0,f=1; char c=getchar();
	while(c<'0'||c>'9') {if(c=='-') f=-1;c=getchar();}
	while(c>='0'&&c<='9') {s=s*10+c-'0';c=getchar();}
	return s*f;
}
struct node{
	LL to,next;
}e[6005];
LL ls[3005],cnt=0;
LL p[3005][3005];
void add(LL x,LL y)
{
	e[cnt]=(node){y,ls[x]};
	ls[x]=cnt++;
	return;
}
LL ksm(LL x,LL y)
{
	LL s=1;
	while(y)
	{
		if(y&1) (s*=x)%=mo;
		(x*=x)%=mo;y>>=1;
	}
	return s;
}
LL size[3005],f[3005][3005],c[3005][3005],g[3005];
LL C(LL n,LL m) {return c[n][m];}
void dfs(LL u,LL fa)
{
	size[u]=1;f[u][1]=1;
	for(LL i=ls[u];~i;i=e[i].next)
	{
		LL v=e[i].to;
		if(v==fa) continue;
		dfs(v,u);
		LL su=size[u],sv=size[v];
		for(LL j=1;j<=su+sv;j++) g[j]=0;
		if(!p[u][v])
		  for(LL i=1;i<=su;i++)
		  {
			LL sum=0;
		    for(LL j=0;j<=sv;j++)
			{
			  (sum+=f[v][j])%=mo;
		      (g[i+j]+=C(i+j-1,i-1)*C(su-i+sv-j,su-i)%mo*f[u][i]%mo*sum%mo)%=mo;
			}		  	
		  }
		else
		  for(LL i=1;i<=su;i++)
		  {
		  	LL sum=0;
		  	for(LL j=1;j<=sv;j++) (sum+=f[v][j])%=mo;
		    for(LL j=0;j<=sv;j++)
			{
			  sum=(sum-f[v][j]+mo)%mo;
		      (g[i+j]+=C(i+j-1,i-1)*C(su-i+sv-j,su-i)%mo*f[u][i]%mo*sum%mo)%=mo;
			}		  	
		  }	
		size[u]+=size[v];
		for(LL j=1;j<=size[u];j++) f[u][j]=g[j]%mo;
	}
	return;
}
int main()
{
	freopen("perm.in","r",stdin);
	freopen("perm.out","w",stdout);
	memset(ls,-1,sizeof(ls));
	LL n=read();
	c[0][0]=1;
	for(int i=1;i<=n;i++)
	{
	  c[i][0]=1;
	  for(int j=1;j<=i;j++) c[i][j]=(c[i-1][j-1]+c[i-1][j])%mo;
	}
	for(LL i=1;i<n;i++) 
	{
		LL u=read(),v=read();
		p[u][v]=1;
		add(u,v);add(v,u);
	}
	dfs(1,0);
	LL ans=0;
	for(LL i=1;i<=n;i++) (ans+=f[1][i])%=mo;
	cout<<ans;
	return 0;
}