夜摘钻石灯

HCY同学又获金牌了！
大家都在庆祝！叶老师决定给HCY发点奖金以祝贺。不过能得到多少奖金，就要看HCY同学的表现了。
叶老师用n-1条线，把n盏钻石灯连起来，每盏灯在夜幕下都闪烁着一个正整数，这个数就是叶老师要给HCY的奖金，不过由于数字太大，最后给的奖金当然不是人民币，而是卢布，还是分。
当HCY选中一盏灯后，跟这盏灯直接连接的灯就会灭掉，他不能选择，他只能选择那些还是亮着的灯。
当然一直在答疑群里答疑的HCY，当然会选到最高的金额，所有他选中灯的数字和就是他的奖金。

输入
第一行包含一个整数 n ，n<=100000。
接下来的一行包含 n 个正整数，第 i 个灯上写的钱数（灯的编号为1~n)。
接下来一共 n-1 行，每行描述n-1条线连着哪些灯，每行2个整数表示2头的灯的编号。
输出
输出一个整数，代表HCY能获得的最大奖金数。
样例输入:
5
1 2 3 4 5
1 2
1 3
2 4
2 5
样例输出:
12

解答：

这个问题是一个典型的树形动态规划问题，问题的本质是在树形结构中找出一个子集，使得这个子集中所有节点的和最大，并且任意两个节点都不相邻。

我们可以通过深度优先搜索（DFS）的方式，从底向上地计算每个节点的最大值。对于每个节点，我们有两种选择：选择该节点或者不选择该节点。如果我们选择了这个节点，那么我们不能选择它的任何子节点。如果我们没有选择这个节点，那么我们可以选择它的所有子节点。

对于每个节点，我们需要维护两个值：f[i][0]表示不选择节点i时的最大值，f[i][1]表示选择节点i时的最大值。这两个值可以通过以下方式计算：

• f[i][0] = ∑max(f[j][0], f[j][1])（对于所有j，j是i的子节点）
• f[i][1] = ∑f[j][0] + a[i]（对于所有j，j是i的子节点）

这样，我们可以使用深度优先搜索的方式，从叶节点开始向上计算，最后的结果就是max(f[root][0], f[root][1])。

下面是C语言实现的代码：

#include <stdio.h>
#include <string.h>
#include <stdlib.h>

#define MAXN 100010
#define ll long long

typedef struct Edge {
    int to;
    struct Edge* next;
} Edge;

Edge* G[MAXN];
ll a[MAXN];
ll f[MAXN][2];
int n;

void add_edge(int u, int v) {
    Edge* edge = (Edge*)malloc(sizeof(Edge));
    edge->to = v;
    edge->next = G[u];
    G[u] = edge;
}

void dfs(int u, int fa) {
    for (Edge* e = G[u]; e; e = e->next) {
        int v = e->to;
        if (v == fa) continue;
        dfs(v, u);
        f[u][0] += f[v][1] > f[v][0] ? f[v][1] : f[v][0];
        f[u][1] += f[v][0];
    }
    f[u][1] += a[u];
}

int main() {
    scanf("%d", &n);
    for (int i = 1; i <= n; ++i) {
        scanf("%lld", &a[i]);
    }
    for (int i = 1; i < n; ++i) {
        int u, v;
        scanf("%d %d", &u, &v);
        add_edge(u, v);
        add_edge(v, u);
    }
    dfs(1, 0);
    printf("%lld\n", f[1][0] > f[1][1] ? f[1][0] : f[1][1]);
    return 0;
}

这段代码首先读入节点数目和每个节点的值，然后读入每条边的信息并构建图。然后从根节点开始进行深度优先搜索，最后输出根节点的最大值。