UVa 1407 树形背包 Caves

这道题可以和POJ 2486 树形背包DP Apple Tree比较着来做。

参考题解

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

const int maxn = 500 + 10;

int n, Q;
vector<int> G[maxn], C[maxn];

int d[maxn][maxn][2], cnt[maxn];

void dfs(int u)
{
    cnt[u] = 1;
    d[u][1][0] = d[u][1][1] = 0;
    for(int i = 0; i < G[u].size(); i++)
    {
        int v = G[u][i];
        int w = C[u][i];
        dfs(v);
        cnt[u] += cnt[v];

        for(int j = cnt[u]; j >= 2; j--)
            for(int x = 0; x <= j && x <= cnt[v]; x++)
            {
                d[u][j][0] = min(d[u][j][0], d[v][x][0] + d[u][j-x][0] + w * 2);
                d[u][j][1] = min(d[u][j][1], d[v][x][0] + d[u][j-x][1] + w * 2);
                d[u][j][1] = min(d[u][j][1], d[v][x][1] + d[u][j-x][0] + w);
            }
    }
}

int main()
{
    int kase = 1;
    while(scanf("%d", &n) == 1 && n)
    {
        for(int i = 0; i < n; i++) { G[i].clear(); C[i].clear(); }
        for(int i = 1; i < n; i++)
        {
            int a, b, d; scanf("%d%d%d", &a, &b, &d);
            G[b].push_back(a); C[b].push_back(d);
        }

        memset(d, 0x3f, sizeof(d));
        dfs(0);

        printf("Case %d:
", kase++);
        scanf("%d", &Q);
        while(Q--)
        {
            int x; scanf("%d", &x);
            int ans;
            for(ans = n; ans > 0; ans--) if(d[0][ans][1] <= x) break;
            printf("%d
", ans);
        }
    }

    return 0;
}