Cod sursa(job #3237806)

Utilizator SilviuC25Silviu Chisalita SilviuC25 Data 13 iulie 2024 11:59:12
Problema Asmax Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.01 kb
#include <bits/stdc++.h>
#define int long long
using namespace std;

ifstream fin("asmax.in");
ofstream fout("asmax.out");

const int MAX_NUM = 16005;

int n, answer = INT_MIN;
vector<bool> visited(MAX_NUM, false);
vector<int> graph[MAX_NUM], v(MAX_NUM);

void dfs(int x) {
    stack<int> s;
    visited[x] = true;
    s.push(x);
    while (!s.empty()) {
        int current = s.top();
        s.pop();
        for (int node : graph[current]) {
            if (!visited[node]) {
                visited[node] = true;
                if (v[node] > 0) {
                    v[x] += v[node];
                }
                s.push(node);
            }
        }
        answer = max(answer, v[x]);
    }
}

int main() {
    fin >> n;
    for (int i = 1; i <= n; ++i) {
        fin >> v[i];
    }
    for (int i = 0; i < n - 1; ++i) {
        int a, b;
        fin >> a >> b;
        graph[a].push_back(b);
        graph[b].push_back(a);
    }
    dfs(1);
    fout << answer;
    return 0;
}