POJ 3659 Cell Phone Network（最小支配集：树形dp）

题目链接：

Cell Phone Network

 

题意：

有一棵树，问最少取树中的多少个节点，能使其与树的其他节点都联通。（一个点只与其有边相连的点联通）

 

思路：

dp[i][state]：以i为根节点，并且其状态为state的情况下，其子树中被染色的点的个数的最小值

dp[i][0]：i不染色，父节点染色覆盖i

dp[i][1]：i不染色，子节点染色覆盖i

dp[i][2]：i染色

=>

对于状态0：该点已被其父节点覆盖，所以它的子节点可以是染色的，也可以是不染色的。

因此：设v为i的子节点，dp[i][0] += min(dp[v][1], dp[v][2]);

对于状态1：该节点需要被其子节点覆盖，因此其子节点中至少要有一个是染色的。

因此：设v为i的子节点，dp[i][1] += min(dp[v][1], dp[v][2])，同时保存Min = min(Min, dp[v][2] - dp[v][1])。

若所有子节点dp[v][1]<dp[v][2]（即所有子节点都不染色），那么dp[root][1] += Min（即强迫一个子节点染色且保证最优解）。

对于状态2：因为该节点本身染色，因此其子节点可以依赖它，也可以依赖它的子节点，也可以其本身就染色。

因此：设v为i的子节点，dp[i][2] += min(min(dp[v][0], dp[v][1]), dp[v][2]);

此外，初始化时dp[i][2]=1，因为其本身染色。

 

另外就是叶子节点无状态1，根节点无状态0。

 

Code：

#include <bits/stdc++.h>
using namespace std;

typedef long long ll;

const int MAX = 10000 + 10;
const int inf = 1e9 + 7;

int n;
vector<int>mp[MAX];
int dp[MAX][3];
/*
dp[i][0]：i不染色，父节点染色覆盖i
dp[i][1]：i不染色，子节点染色覆盖i
dp[i][2]：i染色
dp[i][state]：以i为根节点，并且其状态为state的情况下，其子树中被染色的点的个数的最小值
*/

void dfs(int root, int fa)
{
	//叶节点
	if (root != 1 && mp[root].size() == 1) {
		dp[root][0] = 0;
		dp[root][1] = inf;//叶节点无子节点
		dp[root][2] = 1;
		return;
	}
	int fg = 0, Min = inf;
	dp[root][2] = 1;
	for (int i = 0; i < mp[root].size(); i++) {
		int v = mp[root][i];
		if (v == fa)	continue;
		dfs(v, root);
		dp[root][0] += min(dp[v][1], dp[v][2]);
		dp[root][2] += min(min(dp[v][0], dp[v][1]), dp[v][2]);
		if (dp[v][2] > dp[v][1]) {
			dp[root][1] += dp[v][1];
			Min = min(Min, dp[v][2] - dp[v][1]);
		}
		else {
			fg = 1;
			dp[root][1] += dp[v][2];
		}
	}
	if (!fg) {
		//如果所有子节点dp[v][1]<dp[v][2],则所有子节点不染色
		//这时必须有一个孩子结点染色才可以保证root被覆盖
		dp[root][1] += Min;
	}
}

int main()
{
	scanf("%d", &n);
	for (int i = 1; i < n; i++) {
		int u, v;
		scanf("%d%d", &u, &v);
		mp[u].push_back(v);
		mp[v].push_back(u);
	}
	dfs(1, -1);
	//设节点1为根，且根无父节点
	int ans = min(dp[1][1], dp[1][2]);
	printf("%d\n", ans);
	return 0;
}