HDU1011 Starship Troopers【树形DP】

题目链接：HDU1011 Starship Troopers

题意：n个房间，m个士兵，给出每个房间的bug数和价值，每个士兵可以消灭20个bug，对于每个房间，只有当前的bug被全部消灭才能进入与它相连的房间，如果当前房间没有bug也要留一个士兵，求最多可以获得的总价值；

分析：dp[i][j]表示房间i有j个士兵的价值；

#include<bits/stdc++.h>
#define pb push_back
using namespace std;
const int maxn=105;
vector<int> g[maxn];
int n,m;
int a[maxn],bug[maxn],dp[maxn][maxn],b[maxn];
void dfs(int x,int fa)
{
	for(int i=b[x];i<=m;i++) dp[x][i]=a[x];
	for(auto y:g[x])
	{
		if(y==fa) continue;
		dfs(y,x);
		for(int j=m;j>=b[x]+1;j--)//正着扫会重复
			for(int k=1;k<=j-b[x];k++)
				dp[x][j]=max(dp[x][j],dp[y][k]+dp[x][j-k]);
	}
}
void rua()
{
	if(n==-1 && m==-1) return;
	for(int i=0;i<=n;i++) g[i].clear();
	memset(dp,0,sizeof(dp));
	for(int i=1;i<=n;i++) scanf("%d%d",&bug[i],&a[i]),b[i]=(bug[i]+19)/20;
	for(int i=1;i<n;i++)
	{
		int x,y;scanf("%d%d",&x,&y);
		g[x].pb(y);g[y].pb(x);
	}
	if(!m) {puts("0");return;}
	dfs(1,1);
	printf("%d\n",dp[1][m]);
}
int main()
{
	while(~scanf("%d%d",&n,&m)) rua();
	return 0;
}