Pagini recente » Monitorul de evaluare | Cod sursa (job #3361876) | Cod sursa (job #3362196) | Rating holtea andreea (reeaandreea) | Cod sursa (job #3361980)
// 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;
}