没有上司的舞会

题目描述

某大学有N个职员，编号为1~N。他们之间有从属关系，也就是说他们的关系就像一棵以校长为根的树，父结点就是子结点的直接上司。现在有个周年庆宴会，宴会每邀请来一个职员都会增加一定的快乐指数Ri，但是呢，如果某个职员的上司来参加舞会了，那么这个职员就无论如何也不肯来参加舞会了。所以，请你编程计算，邀请哪些职员可以使快乐指数最大，求最大的快乐指数。

输入输出格式

输入格式：

 

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

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

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

最后一行输入0 0

 

输出格式：

 

输出最大的快乐指数。

 

输入输出样例

输入样例#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 

该题属于树形DP，利用vector建树，找到没有下级的人，然后从这个人开始DFS；

而每个点都有两种情况，一是选择这个点取得的最大值和不选取得最大值，dp[i][1]表示选择这个点时的最大值， dp[i][0]表示不选这个点取得最大值，方程式为 dp[x][0]+=max(dp[i][1],dp[i][0]);dp[x][1]+=dp[i][0];

#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
int dp[6001][2];
int R[6001];
int boss[6001]= {0};
vector<int> sub[60001];
void DFS(int x) {
	for(int j=0; j<sub[x].size(); j++) {
		int i=sub[x][j];
		DFS(i);
		dp[x][0]+=max(dp[i][0],dp[i][1]);//如果不选x则dp[x][0]的值就应加上选或不选他的下级中的较大值 
		dp[x][1]+=dp[i][0];//如果选了x，则直接他的值就应该加上不选他的下级的值 
	}
	dp[x][1]+=R[x];//最后如果加上他，应加上他自己本身高兴值 
}
int main() {
	int N;
	cin>>N;
	for(int i=1; i<=N; i++) {
		cin>>R[i];
	}
	for(int i=2; i<=N; i++) {
		int a,b;
		cin>>a>>b;
		sub[b].push_back(a);
		boss[a]=b;
	}
	for(int i=1; i<=N; i++) {
		if(boss[i]==0) {
			DFS(i);
				cout<<max(dp[i][0],dp[i][1]); 
			break;
		}
	}

	return 0;

}