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;
}