【树形DP】没有上司的晚会 (ssl 1607)

$$没有上司的晚会$$

Description

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

Input

第一行一个整数N。(1<=N<=6000)

接下来N行，第i+1行表示i号职员的快乐指数Ri。(-128<=Ri<=127)

接下来N-1行，每行输入一对整数L,K。表示K是L的直接上司。

最后一行输入0,0。

Output

输出最大的快乐指数。

Sample Input

7

1

1

1

1

1

1

1

1 3

2 3

6 4

7 4

4 5

3 5

0 0

Sample Output

5

题目大意：

有一场party，有n个人会来，每个人都有自己的happy度，部分人有自己上司（见到自己的上司就会无法发挥happy值），求最大的happy指数

解题方法：

用f[i][1]（f[i][1]+=f[i的下属][0]）表示第i个人去，f[i][0]（f[i][0]+=max（f[i的下属][0]，f[i的下属][1]））表示第i个人不去，用链表的形式存每一条关系线的上司（x），属下（y），头相等的另一条线（next），再用一个head来存每个点连接下属的线，再用递归来求出结果。

#include<cstdio>
#include<iostream>
using namespace std;
int n,k,l,pd[6002],a[6002],f[6002][3],head[6002];
struct abc
{
	int x,y,next;
}s[6002];
void dp(int dep)
{
	f[dep][1]=a[dep];
	for (int i=head[dep];i;i=s[i].next)
	{
		dp(s[i].y);//递归
		f[dep][0]+=max(f[s[i].y][1],f[s[i].y][0]);
		f[dep][1]+=f[s[i].y][0];
	}
}
int main()
{
	scanf("%d",&n);
	for (int i=1;i<=n;i++)
	  scanf("%d",&a[i]);
	for (int i=1;i<=n;i++)
	  {
	  	scanf("%d%d",&l,&k);
	  	s[i].x=k;//存上司
	  	s[i].y=l;//存下属
	  	s[i].next=head[k];//把当前的next指向上一个head
	  	head[k]=i;//然后替代掉head
	  	pd[l]=1;//有上司
	  }
	for (int i=1;i<=n;i++)
	  if (!pd[i])//没上司的
	    {
	    	dp(i);//递归
	    	printf("%d",max(f[i][1],f[i][0]));//去和不去中挑最大的
	    	break;
	    }
}