Pagini recente » Cod sursa (job #381499) | Cod sursa (job #3161899) | Cod sursa (job #2703191) | Cod sursa (job #1150491) | Cod sursa (job #1627125)
#include <iostream>
#include <cstdio>
#include <vector>
#define MAXN 16050
using namespace std;
int n;
int val[MAXN], par[MAXN];
vector<int> graf[MAXN];
void citire()
{
scanf("%d", &n);
for (int i = 1; i <= n; i++)
scanf("%d", &val[i]);
for (int i = 1; i <= n-1; i++) {
int a, b;
scanf("%d %d", &a, &b);
graf[a].push_back(b);
graf[b].push_back(a);
}
}
int solve(int nod)
{
int sum = val[nod];
for (int i = 0, t = graf[nod].size(); i < t; i++) {
if (par[graf[nod][i]]) continue;
par[graf[nod][i]] = nod;
int subarb = solve(graf[nod][i]);
if (subarb > 0) sum += subarb;
}
return sum;
}
int main()
{
freopen("asmax.in", "r", stdin);
freopen("asmax.out", "w", stdout);
citire();
par[1] = 1;
printf("%d\n", solve(1));
return 0;
}