Cod sursa(job #2469915)

Utilizator etienAndrone Stefan etien Data 8 octombrie 2019 12:04:41
Problema Asmax Scor 20
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.78 kb
#include <fstream>
#include <iostream>
#include <algorithm>

using namespace std;

ifstream fin("asmax.in");
ofstream fout("asmax.out");

int n,i,nod[16001],p;
int a,b;
int lst[16001],urm[32001],vf[16001];
int viz[16001];
void dfs(int x)
{
    viz[x]=true;
    for(int p=lst[x]; p!=0; p=urm[p])
    {
        int y=vf[x];
        if(!viz[y])
        {
            dfs(y);
            if(nod[y]>0)
                nod[x]+=nod[y];
        }
    }
}

void adauga(int x,int y)
{
    vf[++p]=y;
    urm[p]=lst[x];
    lst[x]=p;
}
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);
    return 0;
    fin.close();
    fout.close();
}