hdu 1561 The more, The Better (树形dp)

题意:每个城堡有一定价值的财宝,可以选择攻打m个城市。但是攻打有些城市之前一定要先

攻打其他的城市。现在已知每个城堡中财宝的价值和城市间的依赖关系。求选择攻打m个城市

能获得的最大价值。


分析:

虽然题目没有明确指出。但是很明显这是一个存在依赖关系的背包问题,而且是依赖中有依赖(可能攻打

a必须先攻打b,而攻打b又要先攻打c)

我们把这种依赖关系用树表示,必须先攻打的是父亲,后攻打的是他的儿子。

dp[i][j]表示选择以i节点为根的子树中j个节点得到的最大财富值。

这样就有两种方案,分别对应了两个子问题:

1】选择以i的第x个儿子t为根的子树中的j-k个节点,然后选择以i为根的除x为根的子树外的其他子节点

(包括它自己)的k个节点。(注意这里出现了子树中的子树的概念,就是有j-k个节点是集中在子树的

子树中选)

2】直接在以i为根的树中(只涉及到i的下一层,也就是i的儿子)选j个节点。

我们已经知道dp[i][1]=val(输入的财宝价值),因为选1个一定会选自己。


依赖关系形成森林,进行树形dp前要把森林转化为树,所以

要加一个根节点:0。结果就是dp[0][m+1] (包括增加的根节点一共m+1个点)


#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;

int mp[210][210],num[210],dp[210][210];
int vis[210],n,m;

int maxx(int x,int y)
{
    return x>y?x:y;
}

void dfs(int r)
{
    vis[r]=1;
    for(int i=1;i<=num[r];i++)
    {
        int t=mp[r][i];
        if(!vis[t])
            dfs(t);
        for(int j=m;j>=2;j--)
        {
            for(int k=1;k<j;k++)
            {
                if(dp[t][j-k]!=-1 && dp[r][k]!=-1)
                    dp[r][j]=maxx(dp[r][j],dp[t][j-k]+dp[r][k]);
            }
        }
    }
}

int main()
{
    int a,b;
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        if(n==0 && m==0)
            break;
        memset(num,0,sizeof(num));
        dp[0][1]=0;
        for(int i=1;i<=n;i++)
        {
            scanf("%d%d",&a,&b);
            dp[i][1]=b;
            mp[a][++num[a]]=i;
        }
        m++;
        for(int i=0;i<=n;i++)
        {
            vis[i]=0;
            dp[i][0]=0;
            for(int j=2;j<=m;j++)
                dp[i][j]=-1;
        }
        dfs(0);
        printf("%d\n",dp[0][m]);
    }
    return 0;
}




hdu 1561 The more, The Better (树形dp)

上一篇:PHP FTP操作类( 上传、拷贝、移动、删除文件/创建目录 )


下一篇:maven安装和eclipse集成