CSP-S 例题22-2 周年纪念晚会

描述

Ural 州立大学的校长正在筹备学校的 80 周年纪念聚会。由于学校的职员有不同的职务级别，可以构成一棵以校长为根的人事关系树。每个资源都有一个唯一的整数编号，从 1 到 N 编号，且对应一个参加聚会所获得的欢乐度。为使每个职员都感到快乐，校长设法使每个职员和其直接上司不会同时参加聚会。
你的任务是设计一份参加聚会者的名单，使总欢乐度最高。

输入描述

第一行是一个整数 N；
接下来 N 行对应 N 个职员的欢乐度，第 i 行的一个整数为第 i 个职员的欢乐度 pi​；
接着是学校的人事关系树，每一行格式为 LK ，表示第 K 个职员是第 L 个职员的直接上司，输入以 00 结束。

输出描述

输出参加聚会者获得的最大欢乐度。

样例输入 1 

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

样例输出 1 

5

提示

数据范围与提示

对于 100% 的数据，1≤N≤6000,−128≤pi​≤127。

#include<bits/stdc++.h>
using namespace std;
//链式前向星存储树 
struct node{
	int u,v,nxt;
}edge[20000];
int cnt,head[10010];
void add(int u,int v){
	cnt++;
	edge[cnt].u=u;
	edge[cnt].v=v;
	edge[cnt].nxt=head[u];
	head[u]=cnt;
} 
int root;//记录根节点
int fu[10010];//记录每个节点的父节点，用于找根节点
int value[10010];//记录欢乐值
int dp[10010][2];//DP数组 
void dfs(int u,int fa){
	for(int i=head[u];i;i=edge[i].nxt){
		int v=edge[i].v;
		if(v==fa) continue;
		dfs(v,u);
		dp[u][0]+=max(dp[v][0],dp[v][1]);
		dp[u][1]+=dp[v][0];
	}
}
int main(){
	int n,,u,v;
	cin>>n;
	for(int i=1;i<=n;i++){
		cin>>value[i];
		dp[i][1]=value[i];
	} 
	while(cin>>v>>u){ //输入顺序 子 父 
		if(u==0&&v==0) break;
		fu[v]=u;
		add(u,v);
		add(v,u); 
	}
	//找根节点
	root=1;
	while(fu[root]){
		root=fu[root];
	} 
	dfs(root,0);
	cout<<max(dp[root][0],dp[root][1]);
	return 0;
}


/*
1.确定树根:树中除根节点外，每个节点都有唯一父节点 
子 父 
1  3   fu[1]=3 
2  3   fu[2]=3
6  4   fu[6]=4
7  4   fu[7]=4
4  5   fu[4]=5
3  5   fu[3]=5
6-fu[6]->4-fu[4]->5-fu[5]->0  5就是根节点 

2.每一个节点，去或者不去
dp[N][2]
	dp[i][0]:第i号节点不去
	dp[i][1]:第i号节点去
如果父节点去，子节点不能去
如果父节点不去，子节点可去可不去 
*/