The more, The Better
Time Limit: 6000/2000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)Total Submission(s): 4447 Accepted Submission(s): 2643
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
#include <iostream> #include <cstdio> #include <cstdlib> #include <cstring> #include <string> #include <queue> #include <algorithm> #include <map> #include <iomanip> #define INF 99999999 typedef long long LL; using namespace std; const int MAX=200+10; int n,m,size,val[MAX]; int dp[MAX],head[MAX],temp[MAX]; struct Edge{ int v,next; Edge(){} Edge(int V,int NEXT):v(V),next(NEXT){} }edge[MAX]; void Init(int num){ for(int i=0;i<=num;++i)head[i]=-1; size=0; } void InsertEdge(int u,int v){ edge[size]=Edge(v,head[u]); head[u]=size++; } int dfs(int u,int Id){ int last=0,now=0; dp[Id]=0; for(int i=head[u];i != -1;i=edge[i].next){ int v=edge[i].v; now=dfs(v,Id+last+1)+1; for(int j=last+1;j<=last+now;++j)dp[Id+j]=0; for(int j=last;j>=0;--j){ for(int k=1;k<=now;++k){ dp[Id+j+k]=max(dp[Id+j+k],dp[Id+j]+temp[k-1]+val[v]); } } last+=now; } for(int i=1;i<=last;++i)temp[i]=dp[Id+i]; return last; } int main(){ int a; while(~scanf("%d%d",&n,&m),n+m){ Init(n); for(int i=1;i<=n;++i){ scanf("%d%d",&a,&val[i]); InsertEdge(a,i); } dfs(0,0); printf("%d\n",temp[m]); } return 0; }