51nod 1299 监狱逃离 树形DP

传送门：51nod 1299

题意：中文题。

思路：

//dp[i]表示警察全部放置完成后i节点的状态 
//0 表示逃犯不能到达i节点并且i节点能（经由其子树）到达出口（不经过警察的情况下）
//1 表示逃犯不能到达i节点并且i节点不能（经由其子树）到达出口（不经过警察的情况下）
//2 表示逃犯能到达i节点并且i节点不能（经由其子树）到达出口（不经过警察的情况下） 

容易得知叶节点的初始状态为0，然后我们树形DP从叶节点开始一点点往上逆推，先随便选一个叶子节点当根开始搜索，注意最后要判断一下所选的根的状态，至于状态转移，根据状态之间的关系推一下就好。

关键是dp状态的选取。

代码：

#include<stdio.h>
#include<bits/stdc++.h>
#define ll long long
#define pi acos(-1)
#define MAXN 100010
#define inf 0x3f3f3f3f
using namespace std;
typedef pair<int,int>P;
int ans = 0;
vector<int> mp[MAXN];
int ind[MAXN], dp[MAXN], ban[MAXN];
void dfs(int u, int fa)
{
	int cnt[3] = {0, 0, 0};
	for(int i = 0; i < mp[u].size(); i++)
	{
		int v = mp[u][i];
		if(v == fa) continue;
		dfs(v, u);
		cnt[dp[v]]++;
	}
	if(ban[u])
	{
		ans += cnt[0];
		dp[u] = 2;
	}
	else if(cnt[0] && cnt[2])
	{
		ans++;
		dp[u] = 1;
	}
	else if(cnt[0]) dp[u] = 0;
	else if(cnt[2]) dp[u] = 2;
	else if(cnt[1]) dp[u] = 1;
}
int main()
{
	int n, m, u, v;
	cin >> n >> m;
	for(int i = 0; i < n; i++)
	{
		scanf("%d %d", &u, &v);
		mp[u].push_back(v);
		mp[v].push_back(u);
		ind[u]++;
		ind[v]++;
	}
	for(int i = 0; i < m; i++)
	{
		scanf("%d", &u);
		ban[u] = 1;
		if(ind[u] == 1)
		{
			cout << -1;
			return 0;
		}
	}
	for(int i = 0; i <= n; i++)
	{
		if(ind[i] == 1)
		{
			dfs(i, -1);
			if(dp[i] == 2) ans++;
			cout << ans << endl;
			return 0; 
		}
	}
    return 0;
}