ZOJ 3626 Treasure Hunt I （树形dp）

题目：http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemCode=3626

题意：有一个危险的村庄，这里有时会有怪兽出现，已知怪兽多久会出现一次，怪兽出现之后发现谁不在家，就会把谁杀掉。一个勇士要得到了一张藏宝图并要去找宝藏，他知道村庄之间的距离和每个村庄里有多少宝藏，问他不被怪兽杀掉的话，最多能拿到多少宝藏。给出的n个村庄和n-1条路，村庄两两之间可以到达。

思路：地图是树形的，勇士所在的村庄为根节点，边权为重量，价值在点上，怪兽出现的时间的为容量。dp[u][j] 记录从 u 出发回到 u 用 j 天最多能收集到的宝藏数。

代码：

#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <math.h>
#include <stdlib.h>
#define INF 0x7fffffff
#define MOD 1000000007
#include <vector>
using namespace std;
typedef long long ll;
int n, k, m, v[105], vis[105], dp[105][205];
struct Edge
{
    int to, cost;
}cur;
vector<Edge> G[105];

void dfs(int u)
{
    vis[u] = 1;
    for(int i = 0; i <= m; i++)
        dp[u][i] = v[u];
    //dp[u][0] = v[u];
    for(int i = 0; i < G[u].size(); i++)
    {
        Edge &e = G[u][i];
        if(vis[e.to]) continue;
        dfs(e.to);
        //printf("v=%d\n", e.to);
        for(int j = m; j >= 1; j--)
        {
            for(int k = 0; j - k - 2 * e.cost >= 0; k++)
            {
                dp[u][j] = max(dp[u][j], dp[u][j - k - 2 * e.cost] + dp[e.to][k]);
                //printf("dp[%d][%d]=%d\n", u, j, dp[u][j]);
            }
        }
    }
}


int main()
{
    #ifdef LOCAL
    freopen("dpdata.txt", "r", stdin);
    #endif

    int a, b, c;
    while(scanf("%d", &n) != EOF)
    {
        for(int i = 1; i <= n; i++)
        {
            scanf("%d", &v[i]);
            G[i].clear();
        }
        for(int i = 0; i < n - 1; i++)
        {
            scanf("%d%d%d", &a, &b, &c);
            cur.cost = c; cur.to = a;
            G[b].push_back(cur);
            cur.to = b;
            G[a].push_back(cur);
        }
        scanf("%d%d", &k, &m);
        //m /= 2;
        memset(vis, 0, sizeof(vis));
        memset(dp, 0, sizeof(dp));
        dfs(k);
        printf("%d\n", dp[k][m]);
    }

    return 0;
}