您的位置：首页 > 其它

HDU 1561 ——The more, The Better（有依赖的树形DP）

2014-07-28 00:27 423 查看
题目分析：

攻占城堡所能获得的最多宝物，但是有个限制，要想攻占某个城堡前必须先攻占另一个城堡

思路：

建图，新建一个根节点为0，那么题目就变为要想取得子节点必须先取得它的父亲节点



今天为了解决这个问题，看了下背包九讲中的有依赖的背包，刷了道模板题。

大概思路是：要想取得附件必须先取主件，主件要么取，要么不取，取得话要怎么分配给附件，用01背包处理附件，然后再把每个主件分组背包

树形的依赖主要是附件是个集合，附件还有附件。森林的概念

根据DP的思想，我们先dfs到叶，01背包操作 即可。只是每次dfs后可以攻占的次数就减一

dp[i][j]表示以i为根攻占j次所能获得的最大宝物，则dp[i][1]=value[i]

这里的dp[i][j]是必须取i的意思，只有取了i，才能取i的子节点

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#define M 10000
using namespace std;
struct pp{int v,w,next;}edge[M*2];int tot,root,head[M],n,m;
inline void addedge(int u,int v,int w,int *h){edge[tot].v=v,edge[tot].w=w,edge[tot].next=h[u],h[u]=tot++;}
int dp[210][210],vis[210],value[210];
void dfs(int u,int c)
{
    vis[u]=1;
    dp[u][1]=value[u];
    for(int i=head[u];i!=-1;i=edge[i].next){
        int v=edge[i].v;
        if(vis[v]) continue;
        dfs(v,c-1);       //剩余所能攻占的次数减一
        for(int k=c;k>=0;--k){  //节点所能攻占的次数
            for(int j=0;j<k;++j){  //在所能攻占的次数里找到最优的
                dp[u][k]=max(dp[u][k],dp[u][k-j]+dp[v][j]); //01背包dp[i][j]=max(dp[i][j],dp[i][j-c]+w)                                                             //这里的w就是dp[v][j]
            }
        }
    }
}
int main()
{
    while(scanf("%d%d",&n,&m)!=EOF&&(n||m)){
        memset(head,-1,sizeof head);
        for(int i=1;i<=n;++i){
            int u,w;
            scanf("%d%d",&u,&w);
            addedge(u,i,w,head);
            value[i]=w;
        }
        memset(dp,0,sizeof dp);
        memset(vis,0,sizeof vis);
        dfs(0,m+1);
        printf("%d\n",dp[0][m+1]);
    }
    return 0;

}
内容来自用户分享和网络整理，不保证内容的准确性，如有侵权内容，可联系管理员处理 点击这里给我发消息
标签： 
相关文章推荐
新的分享
章节导航