（树形dp）洛谷 P1325 标准化模板

知识点：各个节点相连，如同普通树一样，有一定的关系。

状态：dp[x][2]：第x个节点取（dp[x][1]）/不取（dp[x][0]）的最大快乐指数

状态转化：上司不参加（在直系子成员中选择参加/不参加的最大值）：dp[x][0]+=max(dp[son][0],dp[son][1]);

                  上司参加（其直系子成员不参加）：dp[x][1]+=dp[son][0];

初始化：每个人作为一个独立的个体，拥有自己初始的快乐指数

#include <iostream>
#include <cmath>
#include <cstring>
using namespace std;

//人数
int n;
//记录是否有父节点
int fa[6005]; 
//每个人的快乐指数
int happy[6005];
//链式前向星存储连接数据
struct bian{
	int end,next;
}b[6005];
int h[6005];
int cnt;
//状态保存
int dp[6005][2];

void add(int start,int end){
	b[cnt].end=end;
	b[cnt].next=h[start];
	h[start]=cnt++;
}

void dfs(int now){
	//状态初始化
	dp[now][1]=happy[now];
	//遍历
	for(int i=h[now];~i;i=b[i].next){
		int end=b[i].end;
		dfs(end);
		dp[now][1]+=dp[end][0];
		dp[now][0]+=max(dp[end][0],dp[end][1]);
	} 
}

int main(){
	cin>>n;
	memset(h,-1,sizeof(h));
	for(int i=1;i<=n;i++){
		cin>>happy[i];
	}
	for(int i=1;i<n;i++){
		int x,y;
		cin>>x>>y;
		fa[x]=1;
		add(y,x);
	}
	int root;
	for(int i=1;i<=n;i++){
		if(!fa[i]){
			root=i;
			dfs(i);
			break;
		}
	}
	cout<<max(dp[root][0],dp[root][1]);
}