Cod sursa(job #970737)

Utilizator Dddarius95Darius-Florentin Neatu Dddarius95 Data 7 iulie 2013 18:10:42
Problema Asmax Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.89 kb
#include <fstream>
#include <vector>
#define Nmax 16099
#define Vizitat 1
#define Nevizitat 0
#define Infinit 1<<30
using namespace std;
ifstream f("asmax.in");
ofstream g("asmax.out");

int n,v[Nmax],C[Nmax],S;
vector <int> a[Nmax];
char viz[Nmax];

void ReadInput()
{
    f>>n;
    for (int i=1; i<=n; i++) f>>v[i];
    for (int i=1; i<=n-1; i++)
    {
        int x,y;
        f>>x>>y;
        a[x].push_back(y); a[y].push_back(x);
    }
}
void dfs(int nod)
{
    viz[nod]=Vizitat; C[nod]=v[nod];
    for (int i=0; i<a[nod].size();i++)
        if (viz[a[nod][i]]==Nevizitat)
        {
            dfs(a[nod][i]);
            if (C[a[nod][i]]>0)C[nod]+=C[a[nod][i]];
        }
    if (C[nod]>S) S=C[nod];
}

void Solve()
{
    S=-Infinit;
    dfs(1);
    g<<S<<'\n';
}

int main()
{
    ReadInput();
    Solve();
    f.close();g.close();
    return 0;
}