Pagini recente » Cod sursa (job #2735686) | Cod sursa (job #2567933) | Cod sursa (job #2521273) | Cod sursa (job #1409304) | Cod sursa (job #833892)
Cod sursa(job #833892)
#include <cstdio>
#include <algorithm>
#include <iostream>
#include <iterator>
#include <sstream>
#include <vector>
using namespace std;
const int MAXN = 16001;
int n;
vector<int> G[MAXN];
int cost[MAXN];
bool viz[MAXN];
int indeg[MAXN];
int max_sum(int node)
{
bool found = false;
viz[node] = true;
int ssum = 0;
vector<int>::iterator neigh;
for (neigh = G[node].begin(); neigh != G[node].end(); ++neigh) {
if (viz[*neigh]) continue;
found = true;
ssum += max(0, max_sum(*neigh));
}
if (!found) {
return cost[node];
} else {
return ssum + cost[node];
}
}
int main()
{
freopen("asmax.in", "r", stdin);
freopen("asmax.out", "w", stdout);
int max_in = 0, snode;
scanf("%d", &n);
for (int i = 1; i <= n; ++i)
scanf("%d", &cost[i]);
for (int i = 0; i < n - 1; ++i) {
int x, y; scanf("%d %d", &x, &y);
G[x].push_back(y);
G[y].push_back(x);
indeg[x]++;
if (indeg[x] > max_in) max_in = indeg[x], snode = x;
}
printf("%d\n", max_sum(snode));
return 0;
}