Cod sursa(job #2428488)

Utilizator PredescuSebastianIonPredescu Sebastian Ion PredescuSebastianIon Data 5 iunie 2019 16:09:39
Problema Asmax Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.72 kb
#include <bits/stdc++.h>

using namespace std;
ifstream fin("asmax.in");
ofstream fout("asmax.out");
int n,s[20000],x,y,sol=-1000000000,marker[20000];
vector <int> v[20000];
void dfs(int nod)
{
    marker[nod]=1;
    for(auto it : v[nod])
    {
        if(!marker[it])
        {
            dfs(it);
            if(s[it]>0)
            {
                s[nod]+=s[it];
            }
        }
    }
}
int main()
{
    fin>>n;
    for(int i=1; i<=n; i++)
        fin>>s[i];
    for(int i=1; i<n; i++)
    {
        fin>>x>>y;
        v[x].push_back(y);
        v[y].push_back(x);
    }
    dfs(1);
    for(int i=1; i<=n; i++)
    {
        sol=max(sol,s[i]);
    }
    fout<<sol<<'\n';
    return 0;
}