(树形DP) hdu 1561
The more, The Better
Time Limit: 6000/2000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 5822 Accepted Submission(s): 3469
Problem Description
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
Author
8600
dp[i][j]表示从 以i为根节点的子树取j个点得到的最大价值
目标状态 dp[0][m+1];
#include<iostream> #include<cstdio> #include<cstring> #include<string> #include<cmath> #include<cstdlib> #include<algorithm> #include<queue> #include<vector> using namespace std; vector<int> e[205]; int n,m,dp[205][205],w[205]; void dfs(int x,int sum) { dp[x][1]=w[x]; for(int i=0;i<e[x].size();i++) { if(sum>1) dfs(e[x][i],sum-1); for(int j=sum-1;j>=1;j--) { int v=j+1; for(int k=1;k<v;k++) { dp[x][v]=max(dp[x][v],dp[x][v-k]+dp[e[x][i]][k]); } } } } int main() { int x; while(scanf("%d%d",&n,&m)!=EOF) { if(n+m==0) break; memset(dp,0,sizeof(dp)); for(int i=0;i<=n;i++) e[i].clear(); for(int i=1;i<=n;i++) { scanf("%d%d",&x,&w[i]); e[x].push_back(i); } dfs(0,m+1); printf("%d\n",dp[0][m+1]); } return 0; }