Laskettelukeskus

#include <bits/stdc++.h>
#define int long long

using namespace std;

void init_code() {
  ios_base::sync_with_stdio(0);
  cin.tie(0);
#ifndef ONLINE_JUDGE
  freopen("input.txt", "r", stdin);
  freopen("output.txt", "w", stdout);
#endif
}

const int MAXN = 1e5 + 1;

vector<int> g[MAXN + 1];
vector<int> c(MAXN + 1, 0);

int dfs(int v) {
    int s = 0;

    for (auto i : g[v]) {
        s += dfs(i);
    }

    return max(c[v], s);
}

void solve() {
    for (int i = 0; i <= MAXN; ++i) {
        g[i].clear();
    }

    int n;
    cin >> n;
    for (int i = 1; i <= n - 1; ++i) {
        int a, b;
        cin >> a >> b;
        g[a].push_back(b);
    }

    for (int i = 1; i <= n; ++i) {
        cin >> c[i];
    }

    int res = dfs(1);
    cout << res << '\n';   
}

signed main() {
  init_code();
  int t = 1;
  // cin >> t;
  while (t--) {
    solve();
  }
  return 0;
}

Test 1

Group: 1, 2

Verdict: ACCEPTED

Test 2

Group: 1, 2

Verdict: ACCEPTED

Test 3

Group: 1, 2

Verdict: ACCEPTED

Test 4

Group: 2

Verdict: ACCEPTED

Test 5

Group: 2

Verdict: ACCEPTED

Test 6

Group: 1, 2

Verdict: ACCEPTED

Test 7

Group: 2

Verdict: ACCEPTED

Test 8

Group: 1, 2

Verdict: ACCEPTED

Test 9

Group: 2

Verdict: ACCEPTED

Test 10

Group: 1, 2

Verdict: ACCEPTED

Test 11

Group: 2

Verdict: ACCEPTED

Test 12

Group: 1, 2

Verdict: ACCEPTED

Test 13

Group: 2

Verdict: ACCEPTED