P1352 没有上司的舞会

题目描述

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

输入输出格式

输入格式：

 

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

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

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

最后一行输入0 0

 

输出格式：

 

输出最大的快乐指数。

 

输入输出样例

总结：

f[x][0]表示这个点不选的最大欢乐度

f[x][1]表示这个点选的最大欢乐度

#include <bits/stdc++.h>

using namespace std;
const int maxn = 100005;
int head[maxn], cnt = 1, r[maxn];

struct Node {
	int v, nxt;
} G[maxn];
void ins(int u, int v) {
	G[cnt] = (Node) {v, head[u]}; head[u] = cnt++;
} int f[maxn][2];
void dp(int x, int fa) {
	    f[x][1] = r[x];
	for (int i = head[x]; i; i = G[i].nxt) {
		int v = G[i].v; if(v == fa) continue;
		dp(v, x); 
		f[x][1] = f[x][1] + f[v][0];
		f[x][0] = f[x][0] + max(f[v][0], f[v][1]);
	}
}
int main() {
	int n;
	scanf("%d", &n);
	for (int i = 1; i <= n; ++i) {
		scanf("%d", &r[i]);
	}
	for (int i = 1, x, y; i <= n - 1; ++i) {
		scanf("%d%d", &x, &y);
		ins(x, y); ins(y, x);
	}
	dp(1, -1);
	printf("%d\n", max(f[1][0], f[1][1]));
	return 0;
}