【树形dp】快乐的聚会

最新推荐文章于 2021-12-27 16:29:50 发布

HEATHEN__

最新推荐文章于 2021-12-27 16:29:50 发布

阅读量285

点赞数

分类专栏：基础知识

本文链接：https://blog.csdn.net/HEATHEN__/article/details/102608079

版权

基础知识专栏收录该内容

2 篇文章 0 订阅

订阅专栏

【基础算法】快乐的聚会

题目描述
你要组织一个由你公司的人参加的聚会。你希望聚会非常愉快，尽可能多地找些有趣的热闹。但是劝你不要同时邀请某个人和他的上司，因为这可能带来争吵。
给定N个人(姓名，他参加聚会的欢乐指数，以及他上司的名字)，编程找到能使欢乐指数之和最大的若干个人。

输入
第一行一个整数N(N＜100)。
接下来有N行，每一行描述一个人的信息，信息之间用空格隔开。姓名是长度不超过20的字符串，欢乐指数是在0到100之间的整数。

输出
第1行：所邀请的人最大的欢乐指数之和

【样例输入】 Copy
5
BART 1 HOMER
HOMER 2 MONTGOMERY
MONTGOMERY 1 NOBODY
LISA 3 HOMER
SMITHERS 4 MONTGOMERY

【样例输出】 Copy

#include<iostream>//happy
#include<vector>
#include<map>
using namespace std;
#define m 105
vector<unsigned int> son[m];
map<string,unsigned int> id;
unsigned int n,st,f[m][2]; 
struct node{string s,fa;unsigned int w;}a[m]; 
void dp(int x)
{
    f[x][0]=0,f[x][1]=a[x].w;
    for(int i=0;i<son[x].size();i++)
    {
        int y=son[x][i];
        dp(y);
        f[x][1]+=f[y][0];
        f[x][0]+=max(f[y][0],f[y][1]);
    }
}
int main()
{
    ios::sync_with_stdio(false);
    cin>>n;
    for(register int i=1;i<=n;i++) cin>>a[i].s>>a[i].w>>a[i].fa,id[a[i].s]=i;
    for(register int i=1;i<=n;i++) son[id[a[i].fa]].push_back(i);
    for(register int i=1;i<=n;i++) if(!id[a[i].fa]) {st=i;break;}
    dp(st);
    cout<<max(f[st][0],f[st][1]);
    return 0;
}