蓝桥杯训练：动态规划——没有上司的晚会

题目描述


　　Ural周立大学的校长正在筹备学校的80周年纪念聚会。由于学校的职员有不同的职务级别，可以构成一棵以校长为根的人事关系树。每个职员都有一个唯一的整数编号（范围在1到N之间），并且对应一个参加聚会所获得的欢乐度。为了使每个参加聚会者都感到欢乐，校长想设法使每个职员和他（她）的直接上司不会同时参加聚会。
　　你的任务是设计一份参加聚会者的名单，使总的欢乐度最高。


输入格式


　　输入的第一行是一个整数N，1<= N <= 6000
　　以下的N行是对应的N个职员的欢乐度（欢乐度是一个从-128到127之间的整数）
　　接着是学校的人事关系树，树的每一行格式如下：
　　< L > < K >
　　表示第K个职员是第L个职员的直接上司。
　　输入以0 0表示结束


输出格式


　　输出：参加聚会者获得的最大总欢乐度


样例数据


样例输入


7
1
1
1
1
1
1
1
1 3
2 3
6 4
7 4
4 5
3 5
0 0


样例输出


5
来源：网上代码加自己修改BUG，源码地址：http://blog.csdn.net/libin56842/article/details/9814455
完成时间：2017年8月23日
作者：何知令

解题思想：找出所有的根节点，再依次访问他所有的根节点，动态的记录该节点代表员工去和不去的高兴度，最终汇合到其根节点上来，最终把根节点加起来便是所求的答案

代码如下：

/*
题目描述

　　Ural周立大学的校长正在筹备学校的80周年纪念聚会。由于学校的职员有不同的职务级别，可以构成一棵以校长为根的人事关系树。每个职员都有一个唯一的整数编号（范围在1到N之间），并且对应一个参加聚会所获得的欢乐度。为了使每个参加聚会者都感到欢乐，校长想设法使每个职员和他（她）的直接上司不会同时参加聚会。
　　你的任务是设计一份参加聚会者的名单，使总的欢乐度最高。

输入格式

　　输入的第一行是一个整数N，1<= N <= 6000
　　以下的N行是对应的N个职员的欢乐度（欢乐度是一个从-128到127之间的整数）
　　接着是学校的人事关系树，树的每一行格式如下：
　　< L > < K >
　　表示第K个职员是第L个职员的直接上司。
　　输入以0 0表示结束

输出格式

　　输出：参加聚会者获得的最大总欢乐度

样例数据

样例输入

7
1
1
1
1
1
1
1
1 3
2 3
6 4
7 4
4 5
3 5
0 0

样例输出

5
来源：网上代码加自己修改BUG，源码地址：http://blog.csdn.net/libin56842/article/details/9814455
完成时间：2017年8月23日
作者：何知令
解题思想：找出所有的根节点，再依次访问他所有的根节点，动态的记录该节点代表员工去和不去的高兴度，最终汇合到其根节点上来，最终把根节点加起来便是所求的答案
*/
#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std;
int father[6005],vis[6005],dp[6005][2],t;
void dfs(int node)
{
    int i;
    vis[node] = 1;
    for(i = 1; i<=t; i++)
    {
        if(!vis[i] && father[i] == node)
        {
            dfs(i);
            dp[node][1]+=dp[i][0];//node去,则i必不能去
            dp[node][0]+=max(dp[i][0],dp[i][1]);//node不去，取i去或不去的最大值
        }
    }
}
int main()
{
    int i,l,k;
    int sum=0;
    scanf("%d",&t);
    for(i = 1; i<=t; i++)
        scanf("%d",&dp[i][1]);
    while(scanf("%d%d",&l,&k),l+k>0)
    {
        father[l] = k;//记录上司
    }
    memset(vis,0,sizeof(vis));
    for(i=1; i<=t; i++)
        if(father[i]==0)
        {
            dfs(i);
            sum+=max(dp[i][1],dp[i][0]);
        }
    printf("%d\n",sum);
    return 0;
}

程序运行结果展示：

知识点总结：动态规划

学习心得：网上代码加改BUG