hdu 4276 The Ghost Blows Light(树型DP)

        题意:给出一棵树,走过每条边都要花费时间,每个点有一定价值的物品。求在给定时间内能不能从1走到n,如果能,最多可以带走多少价值的物品。

        思路:首先对树进行一遍搜索,看能不能从1走到n,如果不能直接输出结果。如果能,记录路径。

        因为这条路径上的每条边都需要且只需走一次,每个点的物品都会被取走。所以以这条路径上的每个点作为父节点,对它所有子节点建立分组背包的模型。即相当于走到每一个路径上的点,都对这个点的所有非路径上的子节点建立分组背包。最后再加上路径本身的价值就是最后要求的结果。

#include<stdio.h>
#include<string.h>
const int N=105;
const int T=505;
struct node
{
    int son;
    int next;
    int cost;
} Edge[N*2];
int head[N],vis[N],dp[N][T],v[N],path[N];
int cnt,k;
int Max(int x,int y)
{
    if(x>y) return x;
    else return y;
}
void AddEdge(int x,int y,int cost)
{
    Edge[cnt].son=y,Edge[cnt].next=head[x],Edge[cnt].cost=cost,head[x]=cnt++;
    Edge[cnt].son=x,Edge[cnt].next=head[y],Edge[cnt].cost=cost,head[y]=cnt++;
    return ;
}
int Find(int father,int n,int cost)
{
    if(father==n) return cost;
    int u=head[father];
    vis[father]=1;
    for(int i=u; i!=-1; i=Edge[i].next)
    {
        int son=Edge[i].son;
        if(vis[son]) continue;
        path[k++]=son;
        int temp=Find(son,n,cost+Edge[i].cost);
        if(temp!=-1) return temp;
        k--;
    }
    return -1;
}
void dfs(int father,int kk,int t,int cost)
{
    if((kk<k&&father!=path[kk]&&cost<=t)||(kk>=k&&cost<=t)) dp[father][cost]=v[father];
    else if(kk<k&&path[kk]==father) memset(dp[father],0,sizeof(dp[father]));
    int u=head[father];
    vis[father]=1;
    for(int i=u; i!=-1; i=Edge[i].next)
    {
        int son=Edge[i].son;
        if(vis[son]) continue;
        dfs(son,kk+1,t,Edge[i].cost*2);
        for(int j=t; j>=0; j--)
        {
            for(int l=0; l<=j; l++)
            {
                if(dp[father][j-l]!=-1)
                    dp[father][j]=Max(dp[father][j],dp[father][j-l]+dp[son][l]);
            }
        }
    }
    return ;
}
int main()
{
    int n,t;
    while(scanf("%d%d",&n,&t)!=EOF)
    {
        memset(head,-1,sizeof(head));
        cnt=0;
        for(int i=1; i<n; i++)
        {
            int x,y,k;
            scanf("%d%d%d",&x,&y,&k);
            AddEdge(x,y,k);
        }
        for(int i=1; i<=n; i++)
            scanf("%d",&v[i]);
        k=0;
        path[k++]=1;
        memset(vis,0,sizeof(vis));
        int temp=Find(1,n,0);
        if(temp>t)
        {
            printf("Human beings die in pursuit of wealth, and birds die in pursuit of food!\n");
            continue;
        }
        t-=temp;
        memset(dp,-1,sizeof(dp));
        memset(vis,0,sizeof(vis));
        dfs(1,0,t,0);
        for(int i=0; i<k; i++)
            dp[1][t]+=v[path[i]];
        printf("%d\n",dp[1][t]);
    }
    return 0;
}



  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值