标签：highlight main printf amp 接下来 can IV 表示 str
某大学有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;
}
标签：highlight main printf amp 接下来 can IV 表示 str
原文地址：https://www.cnblogs.com/oi-forever/p/9144382.html
