Cod sursa(job #2794358)

Utilizator Edyci123Bicu Codrut Eduard Edyci123 Data 4 noiembrie 2021 18:25:32
Problema Asmax Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.73 kb
#include <bits/stdc++.h>
#define INF 0x3f3f3f3f
#define DIM 16005

using namespace std;

ifstream f("asmax.in");
ofstream g("asmax.out");

int n, v[DIM], maxi = -INF;
vector <int> edges[DIM];
bitset <DIM> used;

void dfs(int nod)
{
    used[nod] = 1;
    for(auto child : edges[nod])
        if(!used[child])
        {
            dfs(child);
            v[nod] = max(v[nod], v[nod] + v[child]);
        }
    maxi = max(maxi, v[nod]);
}

int main()
{
    f >> n;
    for(int i = 1; i <= n; i++)
        f >> v[i];

    for(int i = 1; i < n; i++)
    {
        int x, y;
        f >> x >> y;
        edges[x].push_back(y);
        edges[y].push_back(x);
    }

    dfs(1);

    g << maxi;
    return 0;
}