Starship Troopers HDU - 1011 （树形DP）

Starship Troopers

 题目链接：HDU - 1011 

题意：星际战争中虫子时很烦人的，有你带领的阿尔法舰队要开赴洞穴消灭虫子，舰队上共m个士兵，每个士兵可以消灭20只虫子，洞穴入口是1号房间，洞内有N个房间，每个房间相互连通，每个房间有x个虫子，还有y个首脑，要活捉首脑必须消灭房间内的全部虫子，如果有20只虫子，就需要一个士兵，21只虫子就需要2个士兵；没经过一个房间就必须留下足够的士兵留守，且不能走重复的路；问最多能活捉多少首脑？

思路：dp[i][j]表示i房间留下j名士兵获得的最大价值；j必须>=bug[i]/20+(bug[i]%20?1:0);

#include <bits/stdc++.h>
using namespace std;
int N, M, bug[110], val[110], dp[110][110], vis[110];
vector<int> vec[110];
void dfs(int root){
	int cost = bug[root]/20+(bug[root]%20?1:0);//计算出root房间最少要留下的士兵
	for(int i=cost; i<=M; i++)
		dp[root][i]=val[root];
	vis[root]=1;
	for(int i=0; i<vec[root].size(); i++){
		int son=vec[root][i];
		if(!vis[son]){
			dfs(son);
			for(int j=M; j>=cost; j--){
				for(int k=1; k+j<=M; k++){//k表示son房间留下的士兵数；k从1开始，从0开始是没有意义的，但是不知道k从0开始什么错，不知有何影响，按说dp[son][0]是0，不该有影响的，这里一直百思不得其解，只能先理解为k=0时没有意义；
					dp[root][k+j]=max(dp[root][k+j], dp[root][j]+dp[son][k]);
				}
			}
		}
	}
}
int main(){
	while(scanf("%d%d", &N, &M), N!=-1&&M!=-1){
		for(int i=0; i<=N; i++) vec[i].clear();
		memset(dp, 0, sizeof(dp));
		memset(vis, 0, sizeof(vis));
		for(int i=1; i<=N; i++){
			scanf("%d%d", &bug[i], &val[i]);
		}
		for(int i=1; i<N; i++){
			int a, b;
			scanf("%d%d", &a, &b);
			vec[a].push_back(b);
			vec[b].push_back(a);
		}
		if(M==0){
			printf("0\n");
			continue;
		}
		dfs(1);
		printf("%d\n", dp[1][M]);
	}
	return 0;
}