ZOJ 3626 Treasure Hunt I

一道树形DP+01背包的题。用dp[k][j]表示以k的根的树用j天最多可以得到多少宝藏，

m是天数限制，因为要往返，所以将m/2以后就可以直接计算单程的值了。

#include <stdio.h>

#include <string.h>

#include <stdlib.h>

#include <algorithm>



///dp[k][j]表示以k的根的树用j天最多可以得到多少宝藏

using namespace std;

const int MAXN = 105;

int e;

int n, k, m;

int first[MAXN], next[MAXN << 1], v[MAXN << 1];

bool vis[MAXN];

int t[MAXN][MAXN], val[MAXN], dp[MAXN][MAXN];



void addedge(int x, int y)

{

    v[e] = y, next[e] = first[x];

    first[x] = e ++;

}



void ReadGraph()

{

    for(int i = 1; i <= n; i ++)

        scanf("%d", &val[i]);

    e = 0;

    memset(vis, false, sizeof vis);

    memset(first, -1, sizeof first);

    for(int j = 0; j < n - 1; j ++)

    {

        int ui, vi, di;

        scanf("%d%d%d", &ui, &vi, &di);

        t[ui][vi] = t[vi][ui] = di;

        addedge(ui, vi);

        addedge(vi, ui);

    }

    scanf("%d%d", &k, &m);

    m /= 2;

}



void dfs(int k)

{

    int i, j, x, s;

    vis[k] = true;

    for(s = first[k]; s != -1; s = next[s])

    {

        x = v[s];

        if(!vis[x])

        {

            dfs(x);

            for(j = m; j >= t[k][x]; j --) //从k有足够的时间去x点

                for(i = 0; i <= j - t[k][x]; i ++)

                {

                    dp[k][j] = max(dp[k][j], dp[k][j - t[k][x] - i] + dp[x][i]);

                }

        }

    }

}



int main()

{

    while(scanf("%d", &n) == 1)

    {

        ReadGraph();

        for(int i = 1; i <= n; i ++)

            for(int j = 0; j <= m; j ++)

                dp[i][j] = val[i];

        dfs(k);

        printf("%d\n", dp[k][m]);

    }

    return 0;

}