HDU1520——Anniversary party(树形DP)

题意:有n个人,他们之间有上司和下属关系,每个人有自己的价值,现在要选一部分人使其满足上司和下属不同时被选到的情况下价值总和最大。更直接的讲就是,在一棵树中选价值总和尽量多的节点但是不能同时选到一个节点和它的直接父节点。

题解:因为这里要求选的点的个数是不限定的,只需要满足价值总和尽量大。而对于每个点来说,只有选和不选两种状态,这种情况其实和01背包有一些类似,所以我们可以用背包的思想来做。

定义dp[i][0]表示节点i不选,dp[i][1]表示节点要选。初始化dp[i][0]为0(不选这个点一开始就没有价值),dp[i][1]=a[i](选了这个点肯定它的价值就要加到总的价值中去)。那么状态转移时:

不选当前这个点i:dp[i][0]+=max(dp[j][0],dp[j][1])//当前节点不选那么它的儿子可以任意选不选。

选当前这个点i:dp[i][1]+=dp[i][0]//当前节点已经被选那么它的儿子就不能和它同时被选。

而最终的答案由根节点的最大值决定:ans=max(dp[rt][0],dp[rt][1]);

整个过程十分简单,在推出了状态转移方程以后只需要根据给的信息建树然后在树上dfs即可得到答案,需要注意的是为了找出根节点一开始要记录每个节点的入度,最后入度为0的节点就是我们的根节点。

附上代码:

#include<iostream>
#include<stdio.h>
#include<algorithm>
#include<string.h>
#include<math.h>
#include<vector>
using namespace std;
const int maxn=6010;
int dp[maxn][2];
int a[maxn],in[maxn],vis[maxn];
vector<int>q[maxn];
void dfs(int st)
{
   vis[st]=1;
    int num=q[st].size();
    for(int i=0;i<num;i++)
    {
        int nn=q[st][i];
       if(vis[nn]) continue;
       dfs(nn);
        dp[st][0]+=max(dp[nn][0],dp[nn][1]);
       dp[st][1]+=dp[nn][0];
    }
    return;
}
int main()
{
    int n,x,y;
    while(scanf("%d",&n)!=EOF&&n){
    for(int i=1;i<=n;i++)
        scanf("%d",&a[i]);
        memset(in,0,sizeof(in));
        for(int i=1;i<=n;i++)
            q[i].clear();
   while(scanf("%d%d",&x,&y)!=EOF&&x+y){
        q[y].push_back(x);
        in[x]++;
    }
    memset(dp,0,sizeof(dp));
    for(int i=1;i<=n;i++)
       dp[i][1]=a[i];
    for(int i=1;i<=n;i++)
    {
        if(!in[i])//入度为0那就是根节点
        {
            memset(vis,0,sizeof(vis));
            dfs(i);
            printf("%d\n",max(dp[i][0],dp[i][1]));
            break;
        }
    }
    }
    return 0;
}
 

猜你喜欢

转载自blog.csdn.net/wookaikaiko/article/details/81092410