Cod sursa(job #3294724)

Utilizator _andr31Rusanescu Andrei-Marian _andr31 Data 27 aprilie 2025 17:58:30
Problema Asmax Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.31 kb
#include <bits/stdc++.h>
using namespace std;

int main() {
    ifstream fin("asmax.in");
    ofstream fout("asmax.out");
    int n;
    fin >> n;

    vector<long long> dp(n + 1);
    // vector<long long> in_degree(n + 1);
    for (int i = 1; i <= n; ++i) {
        fin >> dp[i];
    }

    vector<vector<int>> adj(n + 1);
    for (int i = 1, x, y; i < n; ++i) {
        fin >> x >> y;
        adj[x].push_back(y);
        adj[y].push_back(x);
        // ++in_degree[y];
        // ++in_degree[x];
    }

    long long sol = LONG_MIN;
    // function<void(int)> dfs = [&](int nod) -> void {
    //     for (int vec : adj[nod]) {
    //         --in_degree[vec];
    //         if (!in_degree[vec]) {
    //             dfs(vec);
    //             dp[nod] = max(dp[nod], dp[nod] + dp[vec]);
    //         }
    //     }
    //     sol = max(sol, dp[nod]);
    // };
    
    // for (int i = 1; i <= n; ++i) {
    //     if (!in_degree[i]) {
    //         dfs(i);

    //     }
    // }

    function<void(int, int)> dfs = [&](int nod, int parinte) -> void {
        for (int vec : adj[nod]) {
            if (vec != parinte) {
                dfs(vec, nod);
                dp[nod] = max(dp[nod], dp[nod] + dp[vec]);
            }
        }
        sol = max(sol, dp[nod]);
    };
    dfs(1, 0);

    fout << sol << '\n';
    return 0;
}