hdu 1011 Starship Troopers(树形dp)

1、http://acm.hdu.edu.cn/showproblem.php?pid=1011

2、题目大意：（简单树形dp,得好好看看）

你是星河战队的领导者，被送往摧毁小虫基地，这个基地被建在地下室，实际上是一个巨大的洞穴，包含许多由隧道连接的小房间。每个小房间都被一些小虫占领，而且他们的智力藏在一些房间中。你的任务就是摧毁这些小虫跟基地，尽量获得更多的智力。

现在有一张地图，每个房间有智力的可能性，洞穴的结构像一棵树，从入口只有一条路径可以到达其他房间，每经过一个房间，都要留下足够的兵来消灭小虫，而且士兵不可以走重复的房间，即每个房间只能走一次。

一个兵可以消灭20只小虫，因为没有足够的兵力，所以只能挑选几个房间进去消灭小虫，并且获得他们的智力，要求在有限的兵力情况下，获得的最大智力。

输入包括多组样例，每个样例的第一行有两个整数n,m,分别代表房间数和你拥有的兵数，接下来n行每行有两个数，分别代表房间内的小虫数和包含智力的可能性，接下来的n-1行，给出隧道的信息，即房间相连的关系，room1是洞穴的入口。样例以-1，-1 结束 输出：每个测试样例输出一行，输出用有限的兵可以得到智力的最大可能性

测试样例：

5 10

50 10

40 10

40 20

65 30

70 30

1 2

1 3

2 4

2 5

1 1

20 7

-1 -1

3、代码：

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
int map[110][110];
int dp[110][110];
int w[110];
int v[110];
int visit[110];
int n,m;

void dfs(int root)
{
visit[root]=1;
int num=(w[root]+19)/20;
for(int i=num; i<=m; i++)
dp[root][i]=v[root];
for(int i=1; i<=map[root][0]; i++)
{
int u=map[root][i];
if(visit[u]==1)
continue;
else
{
dfs(u);
for(int j=m; j>=num; j--)
{
for(int k=1; j+k<=m; k++)
{
if(dp[u][k])
dp[root][j+k]=max(dp[root][j+k],dp[root][j]+dp[u][k]);
}
}
}
}
}
int main()
{
int a,b;
while(scanf("%d%d",&n,&m)!=EOF)
{
if(n==-1&&m==-1)
break;
memset(w,0,sizeof(w));
memset(v,0,sizeof(v));
memset(dp,0,sizeof(dp));
memset(visit,0,sizeof(visit));
memset(map,0,sizeof(map));//注意初始化
for(int i=1; i<=n; i++)
{
scanf("%d%d",&w[i],&v[i]);
}
for(int i=1; i<n; i++)
{
scanf("%d%d",&a,&b);
map[a][0]++;//以a为根节点有多少个叶子节点
map[a][map[a][0]]=b;//以a为根节点的第几个叶子节点是b
map[b][0]++;
map[b][map[b][0]]=a;//双向，原理同上
}
if(m==0)
printf("0\n");
else
{
dfs(1);
printf("%d\n",dp[1][m]);
}
}
return 0;
}
/*
5 10
50 10
40 10
40 20
65 30
70 30
1 2
1 3
2 4
2 5
1 1
20 7
-1 -1
*/