Atcoder DP Contest V题解（换根dp）

这是树形 dp 中的换根 dp。

atcoder 题目传送门 & 洛谷题目传送门

---

Description

有一个 $n$ 个节点的树，对一些节点染色，使得所有被染色的节点是一个连通块。求对于 $1,2,...,n$ 每个节点，该节点被染色的方案个数。所有答案对 $M$ 取模。

• $1\le n\le 10^5$

• $2\le M\le 10^9$

---

Solution

换根 dp，我们需要考虑两个 dp 状态：

• $dp{1}_u$ 表示考虑以 $u$ 为根节点的子树，染色节点 $u$ 被染色的方案数。

• $dp{2}_u$ 表示考虑以 $u$ 为根节点的子树外的节点和 $u$，节点 $u$ 被染色的方案数。

转移：

对于换根 dp ，我们要用两次 dfs 转移 dp 值。

第一次 dfs 选取节点 0 为根，在这棵有根树上转移 $dp1$ 的值。

$$dp{1}_u$$