hdu-1561 The more, The Better (树形dp入门,有依赖的背包问题

前端之家收集整理的这篇文章主要介绍了hdu-1561 The more, The Better (树形dp入门,有依赖的背包问题前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。


The more,The Better

Time Limit: 6000/2000 MS (Java/Others)Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 4954Accepted Submission(s): 2922


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
  1. 3 2
  2. 0 1
  3. 0 2
  4. 0 3
  5. 7 4
  6. 2 2
  7. 0 1
  8. 0 4
  9. 2 1
  10. 7 1
  11. 7 6
  12. 2 2
  13. 0 0

Sample Output
  1. 5
  2. 13

解题思路:

以树来存储各城堡之间的依赖关系。

首先说状态表示,dp[i][j]表示在以i为根节点的子树上攻破j个城堡可达到的最大收益。

为方便表示,以0号节点为总的根节点。

接下来使用‘背包九讲’中‘有依赖的背包问题’一节的思路来做。

‘有依赖的背包’是指要选某些物品,必须先选某件物品。这与本题恰好符合。

设j1号、j2号。。。jn号这n件物品依赖i号物品,那么可以将这n+1个物品归为一组,那么这n+1个物品可能有以下的取法:


只取第i号;


取第i号和第j1号;

取第i号和第j1、j2号;

。。。

取第i号和第j1、j2。。。jn号;


取第i号和第j2号;

取第i号和第j2、j3号;

。。。

。。。

取第i号和第jn号;


可是这种取法效率很低,很多多余情况。可以用01背包的方法来进行优化,即对这n+1种物品进行一次01背包,即可得出所有需要考虑的情况(具体参考‘背包九讲’)

不过因为i的第k个子节点jk也可能被jk的子节点所依赖,所以要通过dfs先把jk的工作做好再来处理i


代码

  1. #include<iostream>
  2. #include<cstdio>
  3. #include<cstring>
  4. #include<vector>
  5. #include<algorithm>
  6.  
  7. using namespace std;
  8.  
  9. vector<int> map[205];
  10. int val[205],dp[205][205];
  11. int n,m;
  12. //dp[i][j]:以第i个节点为起始,攻击j次的最大获益
  13.  
  14. inline int bet(int x,int y)
  15. {
  16. if(x>y) return x;
  17. return y;
  18. }
  19. void dfs(int bg,int c)
  20. {
  21. int size=map[bg].size();
  22. dp[bg][1]=val[bg];//只攻击该节点
  23. for(int i=0;i<size;++i)//对于bg的所有子节点做01背包
  24. {
  25. if(c>1) dfs(map[bg][i],c-1);
  26. //若未到最后,则继续向下深搜,先算好将来所需的数据
  27.  
  28. for(int j=c;j>1;--j)//当背包容量为c时,j=1时为只选bg
  29. {
  30. for(int k=0;k<=j-1;++k)
  31. {
  32. //攻击了第i个子节点下的k个节点,那么还剩下j-k次机会攻击其余的
  33. dp[bg][j]=bet(dp[bg][j],dp[bg][j-k]+dp[map[bg][i]][k]);
  34. }
  35. }
  36. }
  37. }
  38.  
  39. int main()
  40. {
  41. while(cin>>n>>m)
  42. {
  43. if(n==0&&m==0) break;
  44.  
  45. memset(val,sizeof(val));
  46. memset(dp,sizeof(dp));
  47. for(int i=0;i<=n;++i) map[i].clear();
  48. int a,b;
  49. for(int i=1;i<=n;++i)
  50. {
  51. scanf("%d%d",&a,&b);
  52. map[a].push_back(i);
  53. val[i]=b;
  54. }
  55.  
  56. dfs(0,m+1);//从‘0’开始
  57. cout<<dp[0][m+1]<<endl;
  58. }
  59. }

猜你在找的设计模式相关文章