Cod sursa(job #2506018)

Utilizator andaraluca2001Anda Epure andaraluca2001 Data 7 decembrie 2019 12:53:17
Problema Asmax Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.79 kb
#include <fstream>

using namespace std;

ifstream in("asmax.in");
ofstream out("asmax.out");
int n,st,dr,lst[16009],urm[32009],vf[32009],nr;
int v[16009];
bool viz[16009];

void adauga(int x, int y)

{

    vf[++nr]=y;

    urm[nr]=lst[x];

    lst[x]=nr;

}

void dfs(int x)
{
    viz[x]=true;

    for(int p=lst[x]; p!=0; p=urm[p])
    {
        int y=vf[p];
        if(!viz[y])
        {

            dfs(y);
            if(v[y]>0) v[x]+=v[y];
        }

    }
}

int main()
{
    in>>n;
    for(int i=1; i<=n; i++) in>>v[i];

    for(int i=1; i<=n-1; i++)
    {
        in>>st>>dr;
        adauga(st,dr);
        adauga(dr,st);


    }

    //for(int i=1; i<=n; i++) if(!viz[i]) dfs(i);
    dfs(1);
    int ans=-16001;

    for(int i=1; i<=n; i++) if(v[i]>ans) ans=v[i];
    out<<ans;

    return 0;
}