Pagini recente » Cod sursa (job #328754) | Cod sursa (job #2567041) | Cod sursa (job #3174643) | Cod sursa (job #778810) | Cod sursa (job #973161)
Cod sursa(job #973161)
#include <iostream>
#include <fstream>
#include <vector>
#include <algorithm>
#define nmax 16005
#define inf (1<<30)
using namespace std;
int best[nmax], val[nmax], n, a, b, radacina, sol = -inf;
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;
}