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]);    }}