poj 2486 Apple Tree 树形DP,树形背包

题意:一个树形结构中,每个节点都有权值,从点1 开始漫游,在最多走K步的情况下,问可得到的最大收益。

做法:或许DP考的就是状态分割和把具体状态用数据结构压缩表示的能力。因为是树的结构,所以每个节点有两种状态:经过之后一去不回,经过之后在回来。分别用0表示回来,1表示不回来。有一个贪心条件:在走过的步相同的条件下,回到原节点得到的利益<=不回到原节点的价值。

听说还有另一种方法,学习中。。。

#include<stdio.h>
#include<string.h>
#define LMT 105
/*强大的状态设计*/
/*求最优的题中,走回头路的次数也是被限制的*/
//树的“回路”肯定是一环套一环。
//自己手动模拟一下回路状态。
//每一个点的回路插入次序可任意
 typedef struct
{
    int u,v,next;
}line;
line e[LMT<<1];
int dp[LMT][205][2],wei[LMT],next[LMT];
int all,n,step;
inline int max(int a,int b)
{
    return a>b?a:b;
}
void insert(int u,int v)
{
    e[all].u=u;
    e[all].v=v;
    e[all].next=next[u];
    next[u]=all++;
    e[all].v=u;
    e[all].u=v;
    e[all].next=next[v];
    next[v]=all++;
}
void dfs(int u,int pre)
{
   for(int i=0;i<=step;i++)//可以原地踏步浪费(只是为了形象。。。,多数情况下这种表示法不易遗漏状态,所以一旦证明可行就用)
    dp[u][i][0]=dp[u][i][1]=wei[u];
    int v,x,j,k;
    for(x=next[u];x!=-1;x=e[x].next)
    if(e[x].v!=pre)
    {
        v=e[x].v;
        dfs(v,u);
        for(j=step;j>=1;j--)
          for(k=1;k<=j;k++)
              {
                 dp[u][j][1]=max(dp[u][j][1],dp[u][j-k][0]+dp[v][k-1][1]);
                  if(k%2==0)
                  {
                   dp[u][j][1]=max(dp[u][j][1],dp[u][j-k][1]+dp[v][k-2][0]);//这里充分体现那个贪心而引发的trick
                   dp[u][j][0]=max(dp[u][j][0],dp[u][j-k][0]+dp[v][k-2][0]);
                  }
              }
    }
}
int main()
{
    int i;
    while(scanf("%d%d",&n,&step)!=EOF)
    {
        for(i=1;i<=n;i++)
        scanf("%d",&wei[i]);
        memset(next,-1,sizeof(next));
        all=0;
        for(i=1;i<n;i++)
        {
            int u,v;
            scanf("%d%d",&u,&v);
            insert(u,v);
        }
        dfs(1,0);
        printf("%d\n",dp[1][step][1]);
    }
    return 0;
}



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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值