没有上司的晚会 题解

没有上司的晚会

题目

Ural大学有N个职员，编号为1~N。他们有从属关系，也就是说他们的关系就像一棵以校长为根的树，父结点就是子结点的直接上司。每个职员有一个快乐指数。现在有个周年庆宴会，要求与会职员的快乐指数最大。但是，没有职员愿和直接上司一起与会。

输入

第一行一个整数 N 。(1<=N<=6000)
接下来 N 行，第 i+1 行表示i号职员的快乐指数 Ri 。( -128 <= Ri <= 127 )
接下来 N-1 行，每行输入一对整数 L , K 。表示 K 是 L 的直接上司。
最后一行输入0,0。

输出

输出最大的快乐指数。

样例

input

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

output

5

问题分析

这是一个有向图，又是邻接表我太难了，我不懂，emmmm
找到没有父节点的节点开始循环

两种情况：
这个人去：ta 的下属都不能去，加 ta 的下属不能去的情况
这个人不去：ta 的下属可去可不去，选快乐值大的加

代码

#include<iostream>
#include<cstdio>
using namespace std;
int n,tot,maxn,x,s1;
int a[6100],s[6100][3],head[6100],p[6100];
struct c{
	int s,x,next;
}f[6100];
void dp(int dep)
{
	s[dep][1]=a[dep];
	for (int i=head[dep];i;i=f[i].next)
	{
		dp(f[i].x);
		s[dep][1]=s[dep][1]+s[f[i].x][0]; //去
		s[dep][0]=max(s[f[i].x][0],s[f[i].x][1])+s[dep][0];  //不去
	}
}
int main()
{
	scanf("%d",&n);
	for (int i=1;i<=n;i++)
	    scanf("%d",&a[i]);
	for (int j=1;j<n;j++)
	{
		scanf("%d%d",&x,&s1);
		tot++;
		f[tot].s=s1;
		f[tot].x=x;
		f[tot].next=head[s1];
		head[s1]=tot; //单向，构建邻接表
		p[x]=1; //标记有父节点
	}
	for (int i=1;i<=n;i++)
	    if (p[i]==0)
	    {
	    	dp(i);
	    	maxn=max(maxn,max(s[i][0],s[i][1])); //选快乐值大的
	    }
	cout<<maxn<<endl;
	return 0;
}