Pagini recente » Cod sursa (job #2394301) | Cod sursa (job #2394693) | Cod sursa (job #2399167) | Cod sursa (job #2724907) | Cod sursa (job #2394511)
#include <bits/stdc++.h>
#define pb push_back
using namespace std;
ifstream fin("asmax.in");
ofstream fout("asmax.out");
int n;
vector <int> g[16002];
int a[16002], v[16002];
int bfs(int s){
int sum=a[s];
v[s] = 1;
queue <int> q;
q.push(s);
while(!q.empty()){
s = q.front();
q.pop();
for(int i = 0; i<g[s].size(); i++){
if(v[g[s][i]] == 0){
v[g[s][i]] = v[s]+1;
if(a[g[s][i]] > 0){
sum += a[g[s][i]];
q.push(g[s][i]);
}
else{
int k = bfs(g[s][i]);
if(sum+k>sum)
sum+=k;
}
}
}
}
return sum;
}
int main(){
int j, mn=-1003;
fin >> n;
for(int i = 1; i<=n; i++){
fin >> a[i];
if(a[i] > mn){
mn=a[i];
j=i;
}
}
int x, y;
for(int i = 0; i<n-1;i++){
fin>>x>>y;
g[x].pb(y);
g[y].pb(x);
}
fout << bfs(j);
}