uvalive4015

题目大意：
一棵n个节点的有根树，树的边有正整数权，表示两个节点之间的距离，你的任务是回答这样的询问，从根节点出发，走不超过x单位的距离，最多能走多少个节点，节点经过多次算一个，对于每次的询问输出：经过节点数最大的值

思路：
树形DP。
树形DP一般都是用三维结构完成的。
dp[i][j][k]表示根节点为i经过了j个节点，类似于01背包，k== 0表示返回，k == 1表示不返回。
所以如果返回的话只有一种情况就是根节点和子树都要返回
如果不返回的话就可能根返回子树不返回或者根不返回子树返回。

代码

#include <iostream>
using namespace std;
#include <cstring>
#include <stdio.h>
#include <vector>

const int maxn = 550;

int dp[maxn][maxn][2];
int n,son[maxn],cnt[maxn];
vector <int> g[maxn];

void dfs(int x) {
    for(int i = 1; i <= n; i++)
        dp[x][i][0] = dp[x][i][1] = 0x3f3f3f3f;
    dp[x][1][0] = dp[x][1][1] = 0;
    son[x] = 1;
    for(int i = 0; i < g[x].size();i += 2) {
        int y = g[x][i],w = g[x][i + 1];
        dfs(y);
        for(int j = son[x];j >0 ; j --) {//这里一定要倒着因为是01背包问题。
            for(int k = 1; k <= son[y]; k++) {
                dp[x][j + k][1] = min(dp[x][j + k][1],dp[x][j][1]+dp[y][k][1] + w * 2);
                dp[x][j + k][0] = min(dp[x][j + k][0],min(dp[x][j][0] + dp[y][k][1] + w * 2,dp[x][j][1] + dp[y][k][0] + w));
            }
        }
        son[x] += son[y];
    }
}

int main() {

    int T = 1;
    while(scanf("%d",&n) != EOF && n) {
        for(int i = 0; i <= n; i++)
            g[i].clear();
        memset(cnt,0,sizeof(cnt));
        for(int i = 1; i < n; i++) {
            int u,v,w;
            scanf("%d %d %d",&u,&v,&w);
            cnt[u]++;
            g[v].push_back(u);
            g[v].push_back(w);
        }
        int root = 0;
        for(int i = 1; i < n; i++)
            if(!cnt[i])
                root = i;
        dfs(root);
        int q,x;
        scanf("%d",&q);
        printf("Case %d:\n",T++);
        while(q--) {
            scanf("%d",&x);
            int ans = 1;
            for(int i = 1; i <= n; i++)
                if(dp[root][i][0] <= x)
                    ans = i;
            printf("%d\n",ans);
        }
    }
    return 0;
}