hdu 4276 树形dp背包

题意：n个点，n-1条边，每条边有权值，每个点有价值 ，问你能否在T时间内从1走到n，如果能，获得的最大价值。

       先用spfa求出1至n的最短路径，并且将1到n最短路径上的边权致零，t减去dis【n】这样以后就可以不用考虑怎样回到n了，就和普通的树形dp一样，

状态方程   dp [ u ] [ j ] = max ( dp [ u ][ j ]  ,dp[ v] [k ] +   dp [u ] [j-k-w ] ); w为边权，v是u的子节点。

#include<cstdio>
#include<cstring>
#include<queue>
#include<iostream>
using namespace std;
#define FF  freopen("Input.txt","r",stdin)
#define mem(x,y) memset(x,y,sizeof(x))
#define ll long long
#define inf 1000020
const int N=120;
struct Graph
{
    int head[N*2],val[N*2],to[N*2],next[N*2],tot;
    void init()
    {
        tot=0;
        mem(head,-1);
    }
    void add(int u,int v,int w)
    {
        to[tot]=v;
        val[tot]=w;
        next[tot]=head[u];
        head[u]=tot++;
    }
}G;
int num[110],p[110],pre[110],vis[110],dis[110];
int dp[110][505],n,t;
void spfa(int u)
{
    int i;
    for(i=1;i<=n;i++) dis[i]=inf;
    dis[u]=0;
    queue<int>Q;
    Q.push(u);
    mem(vis,0);
    vis[u]=1;
    while(!Q.empty())
    {
        u=Q.front();Q.pop();
        vis[u]=0;
        for(i=G.head[u];i!=-1;i=G.next[i])
        {
            int v=G.to[i];
            int w=G.val[i];
            if(dis[v]>dis[u]+w)
            {
                pre[v]=u;
                p[v]=i;
                dis[v]=dis[u]+w;
                if(!vis[v])
                {
                    vis[v]=1;
                    Q.push(v);
                }
            }
        }
    }
    for(i=n;i!=1;i=pre[i])
    {
        G.val[p[i]]=0;
        G.val[p[i]^1]=0;
    }
}
void dfs(int u,int f)
{
    int i,j,k;
    for(i=0;i<=t;i++) dp[u][i]+=num[u];
    for(i=G.head[u];i!=-1;i=G.next[i])
    {
        int v=G.to[i],w=G.val[i]*2;
        if(v==f) continue;
        dfs(v,u);
        for(j=t;j>=w;j--)
           for(k=0;j-k>=w;k++)
             dp[u][j]=max(dp[u][j],dp[v][k]+dp[u][j-k-w]);
    }
}
int main()
{
    //FF;
    int i;
    while(~scanf("%d%d",&n,&t))
    {
        int u,v,w;
        G.init();
        for(i=1;i<n;i++)
        {
            scanf("%d%d%d",&u,&v,&w);
            G.add(u,v,w);G.add(v,u,w);
        }
        for(i=1;i<=n;i++)
           scanf("%d",&num[i]);
        spfa(1);
        if(t<dis[n]) { printf("Human beings die in pursuit of wealth, and birds die in pursuit of food!\n");continue;}
        t=t-dis[n];
        mem(dp,0);
        dfs(1,0);
        printf("%d\n",dp[1][t]);
    }
    return 0;
}