Cod sursa(job #3353524)

Utilizator ax_dogaruDogaru Alexandru ax_dogaru Data 7 mai 2026 20:23:19
Problema Asmax Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.75 kb
#include <bits/stdc++.h>

using namespace std;

ifstream fin("asmax.in");
ofstream fout("asmax.out");

int n, x, y, val[16005], dp[16005], maxi=-1e8;

vector<int> vec[16005];

void dfs(int poz, int par) {
    dp[poz]=val[poz];
    maxi=max(maxi, dp[poz]);
    for(auto vc:vec[poz]) {
        if(vc==par) {
            continue;
        }
        dfs(vc, poz);
        if(dp[vc]>0) {
            dp[poz]+=dp[vc];
            maxi=max(maxi, dp[poz]);
        }
    }
}

int main()
{
    fin >> n;
    for(int i=1; i<=n; i++) {
        fin >> val[i];
    }
    for(int i=1; i<=n-1; i++) {
        fin >> x >> y;
        vec[x].push_back(y);
        vec[y].push_back(x);
    }
    dfs(1, 0);
    fout << maxi << "\n";
    return 0;
}