Cod sursa(job #2568055)

Utilizator cpopescu1Popescu Cristian cpopescu1 Data 3 martie 2020 20:35:18
Problema Asmax Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.71 kb
#include <bits/stdc++.h>
#define maxn 16006
using namespace std;
ifstream fin("asmax.in");
ofstream fout("asmax.out");
vector <int> lista[maxn];
bool viz[maxn];
unordered_map <int,int> v;
int n;
void dfs(int nod)
{
    viz[nod]=1;
    for(auto x:lista[nod])
        if(!viz[x])
        {
            dfs(x);
            if(v[x]>0)
                v[nod]+=v[x];
        }
}
int main()
{
    fin>>n;
    for(int i=1;i<=n;i++)
        fin>>v[i];
    int x,y;
    for(int i=1;i<n;i++)
    {
        fin>>x>>y;
        lista[x].push_back(y);
        lista[y].push_back(x);
    }
    dfs(1);
    int ma=INT_MIN;
    for(int i=1;i<=n;i++)
        ma=max(ma,v[i]);
    fout<<ma;
    return 0;
}