Cod sursa(job #834572)

Utilizator thebest001Neagu Rares Florian thebest001 Data 14 decembrie 2012 18:22:59
Problema Asmax Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.98 kb
#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=-1000000,s=0;
    while (!sume.empty()){
        s=sume.top();
        if (max<s)
            max=s;
        sume.pop();
    }
    out<<max;
    return 0;
}