hdu 1011 Starship Troopers（树上的01背包）

树上面的01背包，挺好的题。刚开始01背包那部分自己yy了一下。。。然后就wa了。。。请教了别人一下后顿悟。。。跟hdu1561有点相似，dp[i][j]表示在i节点，体积为j时获得的最大值。对于一条边(u, v)，状态转移方程为：

dp[u][j] = max(dp[u][j], dp[u][j-k] + dp[v][k])；是不是跟1561一样？没错。。。将k个小兵分配到v子树上去刷怪，剩下的看着办吧。。。

改好之后交上去还是wa。。。无奈看了看discuss，原来这题不加m=0的特判是会wa的，坑。。。

#include<iostream>
#include<algorithm>
#include<vector>
#include<cstdio>
#include<cstring>
using namespace std;

#define clr(a, b) memset(a, b, sizeof(a))
const int maxn = 111;
int dp[maxn][maxn], val[maxn], cost[maxn];
int n, m;
vector<int> G[maxn];

void dfs(int u, int fa)
{
    for(int i=cost[u]; i<=m; i++)   dp[u][i] = val[u];
    int nc = G[u].size();
    for(int i=0; i<nc; i++)
    {
        int v = G[u][i];
        if(v == fa)     continue;
        dfs(v, u);
        for(int j=m; j>=cost[u]; j--)
            for(int k=1; k<=j-cost[u]; k++)
                dp[u][j] = max(dp[u][j], dp[u][j-k] + dp[v][k]);
    }
}
int main()
{
    while(scanf("%d%d", &n, &m) && !(n == -1 && m == -1))
    {
        clr(dp, 0);
        for(int i=0; i<maxn; i++)   G[i].clear();
        int a, b;
        for(int i=1; i<=n; i++)
        {
            scanf("%d%d", &a, &val[i]);
            cost[i] = (a + 19) / 20;
        }
        for(int i=1; i<n; i++)
        {
            scanf("%d%d", &a, &b);
            G[a].push_back(b);
            G[b].push_back(a);
        }
        if(m == 0)
        {
            puts("0");
            continue;
        }
        dfs(1, -1);
        printf("%d\n", dp[1][m]);
    }
}