ZOJ2111 HDU1011 Starship Troopers, 树形DP

经典的树形DP题, 题解请看代码中的注释

/*******************************************************************************
 # Author : Neo Fung
 # Email : [email protected]
 # Last modified: 2012-02-03 18:04
 # Filename: ZOJ2111 HDU1011 Starship Troopers.cpp
 # Description : 树形DP. dp[x][y]表示的是在x点放置y人可以获得多少最多的brain.
 切记哪怕经过的点没有bug, 都需要放置人
 ******************************************************************************/
#ifdef _MSC_VER
#define DEBUG
#endif

#include <fstream>
#include <stdio.h>
#include <iostream>
#include <string.h>
#include <string>
#include <limits.h>
#include <algorithm>
#include <math.h>
#include <numeric>
#include <functional>
#include <ctype.h>
#include <vector>
#define MAX 110
using namespace std;

vector<int> vec[MAX];
int bugs[MAX],brains[MAX];
bool visited[MAX];
int dp[MAX][MAX];

void init(const int &n)
{
	memset(visited,false,sizeof(visited));
	memset(dp,0,sizeof(dp));
	for(int i=0;i<=n;++i)
	{
		vec[i].clear();
	}
}

void add(const int &x,const int &y)
{
	vec[x].push_back(y);
	vec[y].push_back(x);
}

void dfs(const int &u, int m)
{
	visited[u]=true;
  int t=(bugs[u]+19)/20; 
  for(int i=t;i<=m;++i)
    dp[u][i]=brains[u];

	for(size_t i=0;i<vec[u].size();++i)
	{
		int v=vec[u][i];
		if(!visited[v])
		{
			dfs(v,m);

      // 假如到达节点 u 的有 j 人(j>=t),
      // 如果到达后继节点 v 的有k人, 则留在u的有j-k人
      // 则对于任一节点有dp[u][j]=max{dp[u][j-k]+dp[v][k]}; 
      for(int j=m;j>t;--j)
				for(int k=1;j-k>=t;++k)
						dp[u][j] = max(dp[u][j],dp[u][j-k]+dp[v][k]);
		}
	}
}

int main(void)
{
#ifdef DEBUG  
  freopen("../stdin.txt","r",stdin);
  freopen("../stdout.txt","w",stdout); 
#endif  

  int n,m,x,y;

  while(~scanf("%d%d",&n,&m)&& !(n==-1&&m==-1))
  {
	  init(n);
	  for(int i=1;i<=n;++i)
		  scanf("%d%d",&bugs[i],&brains[i]);
	  for(int i=1;i<n;++i)
	  {
		  scanf("%d%d",&x,&y);
		  add(x,y);
	  }
    if(m==0)
    {
      printf("0\n");
      continue;
    }
	  dfs(1,m);
	  printf("%d\n",dp[1][m]);
  }

  return 0;
}