没有上司的舞会(树状dp)(最强讲解没有之一)

最新推荐文章于 2022-11-13 10:26:13 发布

原创最新推荐文章于 2022-11-13 10:26:13 发布 · 788 阅读

3 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#算法

题专栏收录该内容

31 篇文章

订阅专栏

探讨了在Ural大学中，如何在避免邀请直接上司的前提下，选择参会职员以最大化快乐指数总和的问题。通过深度优先搜索和动态规划解决，实现了代码实现与效率优化。

题目直通车
Ural大学有N名职员，编号为1~N。

他们的关系就像一棵以校长为根的树，父节点就是子节点的直接上司。

每个职员有一个快乐指数，用整数 Hi 给出，其中 1≤i≤N。

现在要召开一场周年庆宴会，不过，没有职员愿意和直接上司一起参会。

在满足这个条件的前提下，主办方希望邀请一部分职员参会，使得所有参会职员的快乐指数总和最大，求这个最大值
输出格式
输出最大的快乐指数。

数据范围
1≤N≤6000,
−128≤H[i]≤127

为什么说是最强讲解;
y总视频讲解:https://www.acwing.com/video/471/
详细到爆

以下本人代码,个人认为y总的更好

#include<queue>
#include<iostream>
#include<string.h>
using namespace std;
int zhi[6001];
int biao[6001];
int n;
int dp[6001][2];
int head[6001],tail[6001],lens,tailss[6001];
void tree(int sm,int la)
{tail[lens]=sm;
 tailss[lens]=head[la];
 head[la]=lens;
 lens++;
 
}
 
void dfs(int x)
{dp[x][1]=zhi[x];
for(int i=head[x];;i=tailss[i])
    {
        if(tail[i]!=0)
        dfs(tail[i]);
     dp[x][0]+=max(dp[tail[i]][1],dp[tail[i]][0]);
     dp[x][1]+=dp[tail[i]][0];
if(tail[i]==0)
    break;
 
    }
 
}
 
 
 
 
int main()
{
memset(biao,0,sizeof(biao));
cin>>n;
lens=1;
int a,b;
for(int i=1;i<=n;i++ )
    cin>>zhi[i];
for(int i=1;i<=n;i++)
{cin>>a>>b;
tree(a,b);
biao[a]=1;
 
}
int root;
 
for(int i=1;i<=n;i++)
{if(biao[i]==0)
    {root=i;
 
     break;
 
    }
 
}
 
 dfs(root);
 
    cout<<max(dp[root][1],dp[root][0])<<endl;
 
 
 
}