Cod sursa(job #2823832)

Utilizator UnknownPercentageBuca Mihnea-Vicentiu UnknownPercentage Data 29 decembrie 2021 21:05:04
Problema Asmax Scor 90
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.07 kb
#include <bits/stdc++.h>

using namespace std;

inline void Open(const string Name) {
    #ifndef ONLINE_JUDGE
        (void)!freopen((Name + ".in").c_str(), "r", stdin);
        (void)!freopen((Name + ".out").c_str(), "w", stdout);
    #endif
}

vector <int> Gr[16001];

int val[16001], dad[16001], dp[16001];
int N, ans = INT_MIN, nod, root;

void dfs(int v) {
    dp[v] = val[v];
    for(int to : Gr[v]) {
        dfs(to);
        dp[v] = max(dp[v], dp[v] + dp[to]);
    }
}

void solve() {
    cin >> N;
    for(int i = 1;i <= N;i++)
        cin >> val[i];

    for(int i = 1, a, b;i < N;i++) {
        cin >> a >> b;
        Gr[a].emplace_back(b), dad[b] = a;
    }

    for(int i = 1;i <= N;i++)
        if(!dad[i]) {
            root = i;
            break;
        }

    dfs(root);

    for(int i = 1;i <= N;i++)
        ans = max(ans, dp[i]);
    cout << ans;
}

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);

    Open("asmax");

    int T = 1;
    for(;T;T--) {
        solve();
    }

    return 0;
}