Laskettelukeskus

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

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

int dfs(int a, int cost) {
    if (adj[a].size()==0)
        return cost+wt[a];
    if (adj[a].size() == 1)
        return cost+max( wt[a], dfs(adj[a][0], cost));
    return cost+max( wt[a], dfs(adj[a][0], cost)+dfs(adj[a][1], cost) );
}

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

    FOR (n-1) {
        cin >> a >> b;
        adj[a-1].push_back(b-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: WRONG ANSWER

Test 3

Group: 1, 2

Verdict: WRONG ANSWER

Test 4

Group: 2

Verdict: WRONG ANSWER

Test 5

Group: 2

Verdict: WRONG ANSWER

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: WRONG ANSWER

Test 12

Group: 1, 2

Verdict: ACCEPTED

Test 13

Group: 2

Verdict: WRONG ANSWER