hdu 4276 The Ghost Blows Light 夜

http://acm.hdu.edu.cn/showproblem.php?pid=4276

树形dp   比赛时时间不够了没写 

基本思路：

先求是否能到达   如果能到达求每个子树在一定时间下的最优解

ans[i][j] 表示的是 以 i 的父节点为根的子树  还有 j 时间时的最优解

注意的是 到n的路径必须走无需返回 其他的可以走且需要返回

还有一条路所花时间可能是 0 （这里让我wa了N久呀）

代码及其注释：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <map>
#include <vector>
#include <algorithm>

#define LL long long

using namespace std;

const int N=103;
const int INF=0x5fffffff;
int a[N];//财宝
int ans[N][N*5];//如上所诉
//邻接表
int head[N];
struct node
{
    int j,next;
    int d;
}side[N*2];
int I;
int n;
int limit[N];//为-1 表示没有限制  否则表示是到出口的路径 必须走 而且到 i 的时候时间最少为limit[i]
void build(int i,int j,int d)
{
    side[I].j=j;
    side[I].d=d;
    side[I].next=head[i];
    head[i]=I++;
}
void dfs(int x,int pre)//求出limit 并且去掉多余边
{
    int f=0;
    for(int t=head[x];t!=0;f=t,t=side[t].next)
    {
        int l=side[t].j;
        if(l!=pre)
        {
            dfs(l,x);
            if(limit[l]!=-1)
            limit[x]=limit[l]+side[t].d;
        }else
        {
            if(f==0)
            head[x]=side[t].next;
            else
            side[f].next=side[t].next;
        }
    }
    if(x==n)
    limit[x]=0;
}
int dp(int x,int m)
{
    int l=side[x].j;
    if(ans[l][m]!=-1)//表
    return ans[l][m];
    if(limit[l]!=-1&&side[x].d+limit[l]>m)//此路必走 但时间不够则属于非法的走法 答案负无穷
    {
        ans[l][m]=-INF;return ans[l][m];
    }
    ans[l][m]=0;
    if(l==0)//没有节点
    {return ans[l][m];}
    if(limit[l]==-1)//可以直接到兄弟节点 此节点不走
    ans[l][m]=dp(side[x].next,m);
    if(limit[l]==-1)
    {
        int temp=m-side[x].d*2;//往返
        for(int i=0;i<=temp;++i)
        {
            ans[l][m]=max(ans[l][m],a[l]+dp(head[l],i)+dp(side[x].next,temp-i));
        }
    }else
    {
        int temp=m-side[x].d;//无需往返 时间至少为limit[l]
        for(int i=limit[l];i<=temp;++i)
        {
            ans[l][m]=max(ans[l][m],a[l]+dp(head[l],i)+dp(side[x].next,temp-i));
        }
    }
    return ans[l][m];
}
int main()
{
    //freopen("data.txt","r",stdin);
    int m;
    while(scanf("%d %d",&n,&m)!=EOF)
    {
        memset(head,0,sizeof(head));
        side[0].j=0;
        I=1;
        for(int i=1;i<n;++i)
        {
            int l,r,d;
            scanf("%d %d %d",&l,&r,&d);
            build(l,r,d);
            build(r,l,d);
        }
        for(int i=1;i<=n;++i)
        scanf("%d",&a[i]);
        memset(limit,-1,sizeof(limit));
        dfs(1,-1);
        if(limit[1]>m)
        {
            printf("Human beings die in pursuit of wealth, and birds die in pursuit of food!\n");
            continue;
        }
        memset(ans,-1,sizeof(ans));
        printf("%d\n", a[1]+dp(head[1],m));//第一个节点无需时间
    }
    return 0;
}