Pagini recente » Cod sursa (job #1914710) | Cod sursa (job #2147333) | Cod sursa (job #1449039) | Cod sursa (job #1102114) | Cod sursa (job #834570)
Cod sursa(job #834570)
#include <fstream>
#include <vector>
#include <string.h>
#include <stack>
using namespace std;
ifstream in("asmax.in");
ofstream out("asmax.out");
int cost[16001];
vector<int> liste[16001];
bool viz[16001];
stack<int> sume;
int DFS(int x){
viz[x]=true;
int sum=cost[x];
sume.push(sum);
for (vector<int>::iterator i=liste[x].begin();i!=liste[x].end();i++){
if (!viz[*i]){
int rezultat=DFS(*i);
if (rezultat>0)
sum+=rezultat;
sume.push(sum);
}
}
return sum;
}
int main(){
int n;
in>>n;
int x1,x2;
for (int i=1;i<=n;i++)
in>>cost[i];
for (int i=1;i<=n-1;i++){
in>>x1>>x2;
liste[x2].push_back(x1);
liste[x1].push_back(x2);
}
DFS(1);
int max=0,s=0;
while (!sume.empty()){
s=sume.top();
if (max<s)
max=s;
sume.pop();
}
out<<max;
return 0;
}