Time Limit: 6000/2000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 3149 Accepted Submission(s): 1857
/* HDU 1561 The more, The Better 树形DP + 分组背包 建立一颗数,选了子节点,必选父亲结点。 对于每个结点,它的子结点就是个分组背包 */ #include<stdio.h> #include<algorithm> #include<string.h> #include<iostream> using namespace std; const int MAXN=220; struct Node { int to; int next; }; Node edge[MAXN];//建立的有向数,和结点数一样就可以了 int tol; int head[MAXN];//头结点 int value[MAXN];//每个结点的宝物数量 int dp[MAXN][MAXN];//dp[i][j]表示在以i为根的子树上,攻克j个结点获得的最大价值 void init() { memset(head,-1,sizeof(head)); tol=0; memset(dp,0,sizeof(dp)); } void add_edge(int a,int b) //建立一条a->b的有向边 { edge[tol].to=b; edge[tol].next=head[a]; head[a]=tol++; } int n,m; void dfs(int u) { dp[u][1]=value[u];//选一个肯定选自己这个结点 for(int i=head[u];i!=-1;i=edge[i].next) { int v=edge[i].to; dfs(v); //分组背包,外层是对所有的组(参考背包九讲) for(int k=m;k>=1;k--) //for v <- V to 0 //下面这个循环必须是j<k结束,不能取等号,因为至少需要留一个点来取u点 for(int j=1;j<k;j++) // do for 所有的属于当前组 dp[u][k]=max(dp[u][k],dp[u][k-j]+dp[v][j]); } } int main() { int a,b; while(scanf("%d%d",&n,&m)!=EOF) { if(n==0&&m==0)break; init(); for(int i=1;i<=n;i++) { scanf("%d%d",&a,&b); value[i]=b; add_edge(a,i); } value[0]=0; m++;//虚拟构造了结点0 dfs(0); printf("%d\n",dp[0][m]); } return 0; }