Codeforces Round #564 (Div. 2) D(组合数学/树上DP)

D. Nauuo and Circle

题意：
  给定一棵$n$个节点的树，从$1$到$n$编号，现在你需要玩弄这棵树。问按照顺时针遍历能获得多少种不同的序列。最后的答案对$\%998244353$
思路：
  定义$son[u]$表示$u$的子节点的个数。先固定$1$是这个序列中的第一个，因为这是一个环所以最后的答案要乘上$n$。定义$dp[u]$表示的是以$u$为节点的方案数。
  对于节点$1$,我们可以就是对$son[u]$进行全排列，并将排列方式放在$1$的后面，将$son[u]$全排列一共有$A_{son[u]} ^ {son[u]}$种方案，也就是$son[u]!$种方案，而它的每一个儿子$v$同样也有它们的子节点，所以就得到了式子 $$dp[u] = son[u]! * \prod_{v\in G[u]}{dp[v]}$$
而对于非$1$节点，它们自身也可以进行全排列所以对于非$1$的节点，它们的方案数的表达式是$dp[u] = (son[u] + 1)! * \prod_{v\in G[u]}{dp[v]}$
所以最后我们将两种情况合并就是 $$dp[u] = du[u]! \prod_{v\in G[u]}{dp[v]}$$ 其中$du[u]$表示的是$du[u]$的度

void solve() {
	int n;
	std::cin >> n;
	fac[0] = 1;
	rep(i,1,N) fac[i] = fac[i - 1] * i;
	std::vector<int> G[n + 1], du(n + 1);
	rep(i,0,n - 1) {
		int u, v;
		std::cin >> u >> v;
		G[u].push_back(v), G[v].push_back(u);
		du[v] ++, du[u] ++;
	}

	std::vector<Z> dp(n + 1);

	std::function<void(int, int)> dfs = [&] (int u, int fa) -> void {
		Z res = 1;
		for (auto v : G[u]) {
			if (v == fa) continue;
			dfs(v, u);
			res = res * dp[v];
		}
		dp[u] = fac[du[u]] * res;
	};

	dfs(1, 0);
	Z ans = dp[1] * n;
	std::cout << ans.val() << "\n";
}

__EOF__

本文作者：HoneyGrey
本文链接：https://www.cnblogs.com/Haven-/p/16151690.html
关于博主：评论和私信会在第一时间回复。或者直接私信我。
版权声明：本博客所有文章除特别声明外，均采用 BY-NC-SA 许可协议。转载请注明出处！
声援博主：如果您觉得文章对您有帮助，可以点击文章右下角【推荐】一下。您的鼓励是博主的最大动力！