Cod sursa(job #2473649)

Utilizator etienAndrone Stefan etien Data 13 octombrie 2019 23:44:41
Problema Asmax Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.73 kb
#include<fstream>
#include<algorithm>
using namespace std;
ifstream fin("asmax.in");
ofstream fout("asmax.out");
int vf[32001],nod[16001],urm[32001],lst[16001],i,n,p,a,b;
bool viz[16001];
void adauga(int x,int y)
{
    vf[++p]=y;
    urm[p]=lst[x];
    lst[x]=p;
}
void dfs(int x)
{
    int j;
    viz[x]=true;
    for(j=lst[x];j!=0;j=urm[j])
    {
        if(!viz[vf[j]])
        {
            dfs(vf[j]);
            if(nod[vf[j]]>0)
                nod[x]+=nod[vf[j]];
        }
    }
}
int main()
{
    fin>>n;
    for(i=1;i<=n;i++)
        fin>>nod[i];
    for(i=1;i<=n-1;i++)
    {
        fin>>a>>b;
        adauga(a,b);
        adauga(b,a);
    }
    dfs(1);
    fout<<*max_element(nod+1,nod+n+1);
}