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