Cod sursa(job #3337897)

Utilizator raultimarTimar Raul raultimar Data 30 ianuarie 2026 17:44:37
Problema Asmax Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.6 kb
#include <bits/stdc++.h>

using namespace std;
ifstream in("asmax.in");
ofstream out("asmax.out");
int n,el[16001],i,x,y,sol[16001],mix,viz[16001];
set<int>v[16001];
int dfs(int x){
    int mixloc=0;
    viz[x]=1;
    for(auto i:v[x]){
        if (viz[i]==0) mixloc=max(mix,dfs(i));
    }
    mixloc=mixloc+el[x];
    mix=max(mixloc,mix);
    return mixloc;
}
int main()
{
    in>>n;
    for(i=1;i<=n;i++){
        in>>el[i];
    }
    for(i=1;i<=n-1;i++){
        in>>x>>y;
        v[x].insert(y);
        v[y].insert(x);
    }
    mix=max(mix,dfs(1));
    out<<mix;
    return 0;
}