Cod sursa(job #3361980)

Utilizator adimiclaus15Miclaus Adrian Stefan adimiclaus15 Data 31 iulie 2026 11:38:27
Problema Asmax Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.25 kb
// Source: https://usaco.guide/general/io

#include <bits/stdc++.h>
using namespace std;

const int NMAX = 16000;
int dp[NMAX + 1];
vector<int> G[NMAX + 1];
int val[NMAX + 1];
int niv[NMAX + 1];
vector<int> noduri[NMAX + 1]; //noduri[i] = ce noduri sunt pe nivelul i

void dfs(int node, int parent) {
    for(auto vec : G[node]) {
        if(vec != parent) {
            niv[vec] = niv[node] + 1;
            dfs(vec, node);
        }
    }
}

int main() {
    ifstream cin("asmax.in");
    ofstream cout("asmax.out");
	int n;
    cin >> n;
    for(int i = 1; i <= n; i++) {
        cin >> val[i];
    }
    for(int i = 1; i < n; i++) {
        int x, y;
        cin >> x >> y;
        G[x].push_back(y);
        G[y].push_back(x);
    }
    dfs(1, 0);
    int nivmax = 0;
    for(int i = 1; i <= n; i++) {
        noduri[niv[i]].push_back(i);
        nivmax = max(nivmax, niv[i]);
    }
    for(int nivel = nivmax; nivel >= 0; nivel--) {
        for(auto it : noduri[nivel]) {
            dp[it] = val[it];
            for(auto vec : G[it]) {
                if(dp[vec] > 0) {
                     dp[it] += dp[vec];
                }
            }
        }
    }
    int sol = dp[1];
    for(int i = 2; i <= n; i++) {
        sol = max(sol, dp[i]);
    }
    cout << sol;
}