【動態規劃】通向自由的鑰匙

通向自由的钥匙(key.pas/c/cpp) 
   
通向自由的钥匙被放n个房间里，这n个房间由n-1 条走廊连接。但是每个房间里都有
特别的保护魔法，在它的作用下，我无法通过这个房间，也无法取得其中的钥匙。虽然我可
以通过消耗能量来破坏房间里的魔法，但是我的能量是有限的。那么，如果我最先站在 1
号房间（1 号房间的保护魔法依然是有效的，也就是，如果不耗费能量，我无法通过1号房
间，也无法取得房间中的钥匙） ，如果我拥有的能量为P，我最多能取得多少钥匙？ 
 
输入数据 
  第一行包含两个非负整数，第一个为N，第二个为P。 
  接下来n 行 ，按1~n的顺序描述了每个房间。 第i+1行包含两个非负整数cost和 keys，
分别为第 i件房取消魔法需要耗费的能量和房间内钥匙的数量。 
  接下来 n-1行，每行两个非负整数 x,y，表示x 号房间和y 号是连通的。 
 
输出数据 
  一行一个整数，表示取得钥匙的最大值。 
 
样例 
输入:key.in 
5 5 
1 2 
1 1 
1 1 
2 3 
3 4 
1 2 
1 3 
2 4 
2 5 
 
输出: key.out 
7 
 
数据范围 
  对于 20%的测试数据，有n<=20 
  对于 30%的测试数据，有n<=30 
  对于所有测试数据，有p,n<=100, cost <= Maxint, keys<= Maxint 

一道較為基礎的樹形DP。
首先多叉轉二叉，然後再DP即可。
狀態：用f[i][j]表示i這顆子樹，使用j的能量能得到的最大鑰匙數。
轉移方程：f[i][j] = max(f[son[i]][k] + f[bro[i]][j - k - c[i]] + v[i], f[bro[i]][j])。
讀入邊的時候，首先讀成無向圖，再根據無向圖建樹，否則後果很嚴重；還有某個節點消耗的能量可以為0，所以j = 0並不能當作邊界條件。
ACCode：

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <bitset>

using std::max;

const char fi[] = "key.in";
const char fo[] = "key.out";
const int maxN = 110;
const int maxM = 110;
const int MAX = 0x3fffff00;
const int MIN = -MAX;

struct Edge {int dest; Edge *next; };
Edge *edge[maxN];
int f[maxN][maxM];
int bro[maxN];
int son[maxN];
int c[maxN];
int v[maxN];
int n, m;

  void init_file()
  {
    freopen(fi, "r", stdin);
    freopen(fo, "w", stdout);
  }
  
  void insert(int u, int v)
  {
    Edge *p = new Edge;
    p -> dest = v;
    p -> next = edge[u];
    edge[u] = p;
  }
  
  void readdata()
  {
    scanf("%d%d", &n, &m);
    for (int i = 1; i < n + 1; ++i)
      scanf("%d%d", c + i, v + i);
    for (int i = 1; i < n; ++i)
    {
      int u, v;
      scanf("%d%d", &u, &v);
      insert(u, v);
      insert(v, u);
    }
  }
  
  int DP(int i, int j)
  {
    if (i == 0) return 0;
    if (f[i][j]) return f[i][j];
    if (bro[i])
      f[i][j] = max(f[i][j], DP(bro[i], j));
    if (j >= c[i])
    for (int k = 0; k + c[i] < j + 1; ++k)
    {
      f[i][j] = max(f[i][j], DP(son[i], k)
        + DP(bro[i], j - k - c[i]) + v[i]);
    }
    return f[i][j];
  }
  
  void Build(int u, int Last)
  {
    for (Edge *p = edge[u]; p; p = p -> next)
    {
      int v = p -> dest;
      if (v != Last)
      {
        Build(v, u);
        bro[v] = son[u];
        son[u] = v;
      }
    }
  }

  void work()
  {
    Build(1, 0);
    DP(1, m);
    printf("%d", f[1][m]);
  }
  
int main()
{
  init_file();
  readdata();
  work();
  exit(0);
}