Laskettelukeskus

#include <bits/stdc++.h>

using namespace std;

// Shorter types
typedef long long ll;
typedef unsigned long long ull;
typedef vector<ll> vll;
typedef vector<ull> vull;
typedef pair<ll,ll> pll;
typedef vector<vll> adjList;
typedef vector<vector<pll>> adjwList;

// Macros
#define LOOP(i, start, end) for (auto i = start; i <= end; i++)
#define FE_LOOP(i, collection) LOOP(i, 0, collection.size()-1)
#define SORT(v) sort(v.begin(), v.end())
#define REV_SORT(v) sort(v.rbegin(), v.rend())

// Variables
ll n;
adjList adj;
vll needs;

void read_input(){
    cin >> n;
    needs = vll(n+1);
    adj = adjList(n+1, vll());
    LOOP(i, 0, n-2){
        ll a, b;
        cin >> a >> b;
        adj[a].push_back(b);
        //adj[b].push_back(a);
    }
    LOOP(i, 1, n-1){
        cin >> needs[i];
    }
}

// Libs

// Solution
void solve(){
    vector<bool> visited(adj.size()+1, false);

    queue<pll> q;
    q.emplace(1, needs[0]);
    ll bestSum = needs[0];

    while (!q.empty()){
        ll node = q.front().first;
        ll has = q.front().second;
        q.pop();

        visited[node] = true;

        ll childNeeds = 0;
        for (ll child : adj[node]){
            if (visited[child]) continue;
            childNeeds += needs[child];
            q.emplace(child, needs[child]);
        }
        if (childNeeds > has){
            bestSum += childNeeds - has;
        }
    }

    cout << bestSum << "\n";
}

int main(){
    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    read_input();
    solve();

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

Test 11

Group: 2

Verdict: WRONG ANSWER

Test 12

Group: 1, 2

Verdict: ACCEPTED

Test 13

Group: 2

Verdict: ACCEPTED