HDU - 1561 The more, The Better (树形依赖背包)

ACboy很喜欢玩一种战略游戏，在一个地图上，有N座城堡，每座城堡都有一定的宝物，在每次游戏中ACboy允许攻克M个城堡并获得里面的宝物。但由于地理位置原因，有些城堡不能直接攻克，要攻克这些城堡必须先攻克其他某一个特定的城堡。你能帮ACboy算出要获得尽量多的宝物应该攻克哪M个城堡吗？ 

Input

每个测试实例首先包括2个整数，N,M.(1 <= M <= N <= 200);在接下来的N行里，每行包括2个整数，a,b. 在第 i 行，a 代表要攻克第 i 个城堡必须先攻克第 a 个城堡，如果 a = 0 则代表可以直接攻克第 i 个城堡。b 代表第 i 个城堡的宝物数量, b >= 0。当N = 0, M = 0输入结束。

Output

对于每个测试实例，输出一个整数，代表ACboy攻克M个城堡所获得的最多宝物的数量。

Sample Input

3 2
0 1
0 2
0 3
7 4
2 2
0 1
0 4
2 1
7 1
7 6
2 2
0 0

Sample Output

5
13

             dfs序后进行操作进行dp，dp[i][j]表示取这个点之后，一共取到J个点时的最大利益是多少，通过en数组加速传递。

#include<bits/stdc++.h>
using namespace std;
int n, m, v[205];
vector<int>G[205];
int st[205], en[205], cnt, p[205];
void dfs(int u) {
	p[st[u] = ++cnt] = u;
	for (auto v : G[u])dfs(v);
	en[u] = cnt;
}
int dp[205][205];
int main() {
	while (cin >> n >> m && n + m) {
		cnt = 0;
		memset(dp, 0, sizeof dp);
		for (int i = 0; i <= n; i++)G[i].clear();
		for (int i = 1; i <= n; i++) {
			int a, b;
			cin >> a >> b;
			v[i] = b;
			G[a].push_back(i);
		}
		for (int v : G[0])dfs(v);
		//	cout << st[1] << " " << st[2] << " " << st[3] << "\n";
		for (int i = n; i > 0; i--) {
			for (int j = m; j > 0; j--) {
				dp[i][j] = max(dp[i + 1][j - 1] + v[p[i]], dp[en[p[i]] + 1][j]);
			}
		}
		cout << dp[1][m] << "\n";
	}
	return 0;
}