【树状DP】没有上司的晚会

心累.jpg
新知识掌握的过程不会总是顺利的。

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

输出最大的快乐指数。

---

思路

这****的存储结构,令人抓狂
DP，用类似链表的东东存下职员们之间的关系，用了一大堆数组。。。
。
。
。
已知，如果i职员被选去了晚会，那么他的下属一定不会去。
（但下属的下属是可以去的）
So，
我们就用f[i][1]表示他会去的最大快乐值，
f[i][0]表示他不会去的。
（此快乐值是只考虑他和他的下属的最优情况）
那么f[i][1]=他的快乐指数+他的下属不会去的快乐值。
f[i][0]=他所有属下去或不去的快乐值（大的那个）

因为，i不去，他属下也不一定会去，要是他属下不去所能获得的快乐值大于他属下去的快乐值，那么f[i][0]就加上他属下不去的快乐值。
每个属下都这样判断一遍。

超级乱的代码

#include<cstdio>
#define ST 6001
using namespace std;
int max(int a,int b){   //求最大值
	if(a>b) return a;
	return b;
}
struct N{             
	int n,to,next;
} k[ST];
int head[ST],n,tot=0,a[ST],f[ST][ST]={0};
bool kkk[ST];
void add(int i,int j){         //指针存储
	k[++tot].n=i;
	k[tot].to=j;
	k[tot].next=head[i];
	head[i]=tot;
	kkk[j]=true;
}
void init(){               //输入
	scanf("%d",&n);
	for(int i=1;i<=n;++i)
	  scanf("%d",&a[i]);
	for(int i=1;i<n;++i){
		int j,l;
		scanf("%d%d",&j,&l);
		add(l,j);
	}
}
void cc(int now){        //递归求最优解
	f[now][1]=a[now];
	for(int i=head[now];i;i=k[i].next)   //所有下属走一遍
	{
		cc(k[i].to);
		f[now][1]=f[now][1]+f[k[i].to][0];  //去
		f[now][0]=max(f[k[i].to][1],f[k[i].to][0])+f[now][0];   //不去
	}
}
int main(){
	init();   //输入
	int l;
	for(int i=1;i<=n;++i)      //判断一下根节点，从根节点开始递归
	  if(kkk[i]==false) {
	  	l=i;
	  	break;
	  }
	cc(l);
	if(f[l][1]>f[l][0]) printf("%d",f[l][1]);  //输出老板去或不去的最大值
	else printf("%d",f[l][0]);
}