Cod sursa(job #3252073)

Utilizator adimiclaus15Miclaus Adrian Stefan adimiclaus15 Data 28 octombrie 2024 13:31:20
Problema Asmax Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.82 kb
#include <bits/stdc++.h>

using namespace std;

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

const int NMAX = 16000;
vector<int> G[NMAX + 1];
int vis[NMAX + 1];
int v[NMAX + 1];
int dp[NMAX + 1];

void dfs(int node) {
    vis[node] = 1;
    dp[node] = v[node];
    for(auto next : G[node]) {
        if(!vis[next]) {
            dfs(next);
            if(dp[next] > 0) {
                dp[node] += dp[next];
            }
        }
    }
}

int main() {
    int n;
    f >> n;
    for(int i = 1; i <= n; i++) {
        f >> v[i];
    }
    for(int i = 1; i <= n - 1; i++) {
        int x, y;
        f >> x >> y;
        G[x].push_back(y);
        G[y].push_back(x);
    }
    dfs(1);
    int mx = -1e9;
    for(int i = 1; i <= n; i++) {
        if(dp[i] > mx) {
            mx = dp[i];
        }
    }
    g << mx;
    return 0;
}