Laskettelukeskus

#include <bits/stdc++.h>
#define ll long long
#define FOR(m) for (long i = 0; i<m; i++)
using namespace std;

vector<vector<ll>> adj(int(1e5+1), vector<ll>()); // left, right
vector <ll> wt(int(1e5+1), 0);

ll dfs(int a, int b) {
    ll ans = 0ll;
    for (int v : adj[a]) {
        if (v == b)
            continue;
        ans += dfs(v, a);
    }
    return max(ans, wt[a]);
}

int main() {
    ll n, a, b;
    cin >> n;

    FOR (n-1) {
        cin >> a >> b;
        adj[a-1].push_back(b-1);
        adj[b-1].push_back(a-1);
    }
    FOR (n) {
        cin >> a;
        wt[i] = a;
    }

    cout << dfs(0, 0);
}

/*long solve(long rt) {
    long x = 0;
    if (slopes[rt-1].first != 0)
        x += solve(slopes[rt-1].first);
    if (slopes[rt-1].second != 0)
        x += solve(slopes[rt-1].second);
    return max(wt[rt-1], x);
}

int main() {
    long n, x, y;
    cin >> n;

    FOR(n) {
        slopes.push_back({0,0});
        wt.push_back(0);
    }

    FOR (n-1) {
        cin >> x >> y;
        if (slopes[x-1].first == 0)
            slopes[x-1].first = y;
        else
            slopes[x-1].second = y;
    }

    FOR (n) {
        cin >> wt[i];
    }
    long ans = solve(1);
    cout << ans;
    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