[洛谷 P4084 USACO17DEC] Barn Painting G (树形dp经典)

[洛谷 P4084 USACO17DEC] Barn Painting G

---

题目链接

---

大致题意:

给定一颗N个节点组成的树，3种颜色，其中K个节点已染色，要求任意两相邻节点颜色不同，求合法染色方案数

---

解题思路:

• f[i][j]表示i这个点上色为j是方案数（从下往上/从子节点向父节点更新）

• 即对于所有初始节点，f[i][1],f[i][2],f[i][3]都为1

• 当某个节点被指定上色后，那么该节点另外两种颜色的方案数为0。

  列如:当点x被指定上色 2 时：f[x][1]=0,f[x][3]=0 (因为无法上色1和3）

• 对于每个节点，因为不能于子节点上色相同,即：

f[u][1] = f[u][1] * ((f[v][2] + f[v][3]) % mod) % mod;
f[u][2] = f[u][2] * ((f[v][1] + f[v][3]) % mod) % mod;
f[u][3] = f[u][3] * ((f[v][1] + f[v][2]) % mod) % mod;

• 最后，记得取模和开long long

---

AC代码:

#include <bits/stdc++.h>
#define rep(i, n) for (int i = 1; i <= (n); ++i)
using namespace std;
typedef pair<int, int> PII;
typedef long long ll;
const int N = 1e5 + 10, mod = 1e9 + 7;
int n, m;
vector<int>e[N];
ll f[N][5]; //i点涂j颜色的方案数
void dfs(int u, int fa) {
	//初始化
	for (int i = 1; i <= 3; ++i) {
		if (f[u][i]) {
			for (int j = 1; j < i; ++j)
				f[u][j] = 0;
			break;
		}
		f[u][i] = 1;
	}
	for (auto& v : e[u]) {
		if (v == fa)continue;
		dfs(v, u);
		f[u][1] = f[u][1] * ((f[v][2] + f[v][3]) % mod) % mod;
		f[u][2] = f[u][2] * ((f[v][1] + f[v][3]) % mod) % mod;
		f[u][3] = f[u][3] * ((f[v][1] + f[v][2]) % mod) % mod;
	}
}
int main(void)
{
	cin >> n >> m;
	for (int i = 1; i < n; ++i) {
		int a, b; cin >> a >> b;
		e[a].push_back(b);
		e[b].push_back(a);
	}
	while (m--) {
		int a, b; cin >> a >> b;
		f[a][b] = 1;
	}
	dfs(1, -1);
	ll res = (f[1][1] + f[1][2] + f[1][3]) % mod;
	cout << res << endl;
	return 0;
}