Pagini recente » Cod sursa (job #2483584) | Cod sursa (job #2085336) | Cod sursa (job #350331) | Cod sursa (job #1221925) | Cod sursa (job #973158)
Cod sursa(job #973158)
#include <iostream>
#include <fstream>
#include <vector>
#include <algorithm>
#define nmax 16005
using namespace std;
int best[nmax], val[nmax], n, a, b, radacina, sol;
vector <int> son[nmax];
bool used[nmax], orfan[nmax];
void dfs(int cur) {
best[cur] = val[cur];
used[cur] = true;
for(int i=0; i<son[cur].size(); i++) {
if(!used[son[cur][i]]) dfs(son[cur][i]);
best[cur] += max(0, best[son[cur][i]]);
}
}
int main() {
ifstream f("asmax.in");
ofstream g("asmax.out");
f>>n;
for(int i=1; i<=n; i++) f>>val[i], orfan[i] = true;
for(int i=1; i<n; i++) {
f>>a>>b;
son[a].push_back(b);
orfan[b] = false;
}
for(int i=1; i<=n; i++) if(orfan[i]) radacina=i;
dfs(radacina);
for(int i=1; i<=n; i++) sol = max(sol, best[i]);
g<<sol<<"\n";
return 0;
}