Cod sursa(job #1425267)

Utilizator julia_robertsIuliana Robert julia_roberts Data 27 aprilie 2015 09:34:15
Problema Asmax Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.96 kb
#include <fstream>
#include <vector>
using namespace std;
ifstream in("asmax.in");
ofstream out("asmax.out");

const int Max = 16005;
const int oo =  -(1 << 30);

vector <int> Arb[Max];

int n, Val[Max], Sum[Max], Result= -(1 << 30);
bool viz[Max];

void DFS(int nod)
{
    viz[nod] = true;
    Sum[nod] = Val[nod];

    for(int i=0; i < Arb[nod].size(); ++i)
    {
        int vecin = Arb[nod][i];
        if(!viz[vecin])
        {
            DFS(vecin);
            Sum[nod] = max(Sum[nod], Sum[nod] + Sum[vecin]);
        }
    }

    Result = max(Result, Sum[nod]);
}
void Read()
{
    in>>n;
    for(int i=1;i<=n; i++)
        in>>Val[i];

    for(int i = 1;i < n; ++i)
    {
        int x, y;
        in >> x >> y;
        Arb[x].push_back(y);
        Arb[y].push_back(x);
    }
}

int main()
{
    Read();
    DFS(1);
   /* for(int i = 0; i <= n; ++i)
        Result = max(Result, Sum[i]);*/
    out<<Result;
    return 0;
}