算法板子
算法板子

#include<bits/stdc++.h>
using namespace std;
const int N = 6010;
int happy[N];
int h[N], v[N], ne[N], idx;
int f[N][2];
bool vis[N];
void add(int x, int y) {
    v[idx] = y;
    ne[idx] = h[x];
    h[x] = idx++;
}
void dfs(int u) {
    f[u][1] = happy[u];
    for (int i = h[u]; i != -1; i = ne[i]) {
        int j = v[i];
        dfs(j);
        f[u][1] += f[j][0];
        f[u][0] += max(f[j][1], f[j][0]);
    }
}
int main() {
    int n;
    memset(h, -1, sizeof h);
    cin >> n;
    for (int i = 1; i <= n; ++i) {
        scanf("%d", &happy[i]);
    }
    for (int i = 1; i <= n - 1; ++i) {
        int a, b;
        scanf("%d%d", &a, &b);
        add(b, a);
        vis[a] = true;
    }
    int root = 1;
    while (vis[root]) ++root;
    dfs(root);
    cout << max(f[root][0], f[root][1]);
    return 0;
}
本文为原创文章,欢迎分享,勿全文转载,如果内容你实在喜欢,可以加入收藏夹,说不定哪天故事又继续更新了呢。
本文地址:https://qoogle.top/prom-without-a-boss/
最后修改日期:2020年9月14日

作者

留言

撰写回覆或留言

发布留言必须填写的电子邮件地址不会公开。