vector邻接表版本

#include <iostream>
#include <vector>
#include <cstring>
#include <algorithm>

using namespace std;

const int N = 6010;

int n;
vector<int> adj[N];
int happy[N];
int f[N][2];
bool has_fa[N];

void add(int a, int b) {
    adj[b].push_back(a);
}

void dfs(int u) {
    f[u][1] = happy[u];

    for (int j : adj[u]) {
        dfs(j);

        f[u][1] += f[j][0];
        f[u][0] += max(f[j][0], f[j][1]);
    }
}

int main() {
    scanf("%d", &n);

    for (int i = 1; i <= n; i++) scanf("%d", &happy[i]);

    for (int i = 0; i < n - 1; i++) {
        int a, b;
        scanf("%d%d", &a, &b);
        add(a, b);
        has_fa[a] = true;
    }

    int root = 1;
    while (has_fa[root]) root++;

    dfs(root);

    printf("%d\n", max(f[root][0], f[root][1]));

    return 0;
}

算法1

(暴力枚举) $O(n^2)$

blablabla

时间复杂度

参考文献

C++ 代码

blablabla

算法2

(暴力枚举) $O(n^2)$

blablabla

时间复杂度

参考文献

C++ 代码

blablabla