Cod sursa(job #2487490)

Utilizator eduardcadarCadar Eduard eduardcadar Data 4 noiembrie 2019 20:52:08
Problema Asmax Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.76 kb
#include <fstream>
#include <climits>
#include <vector>
using namespace std;
ifstream f("asmax.in");
ofstream g("asmax.out");
#define nmax 16001
vector<int> a[nmax];
int n, v[nmax], maxi = INT_MIN;
bool use[nmax];
void dfs(int nod) {
    use[nod] = true;
    for (int i = 0; i < a[nod].size(); ++i) {
        if (!use[a[nod][i]]) {
            dfs(a[nod][i]);
            if (v[a[nod][i]] > 0) v[nod] += v[a[nod][i]];
        }
    }
}
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;
        a[x].push_back(y);
        a[y].push_back(x);
    }
    dfs(1);
    for (int i = 1; i <= n; ++i) maxi = max(maxi, v[i]);
    g << maxi << '\n';
    return 0;
}