Laskettelukeskus

#include <bits/stdc++.h>
 
using namespace std;
using ull = uint64_t;
using sll = int64_t;
static const ull inf = UINT64_MAX;
static const sll infs = INT64_MAX;
 
using vu = vector<ull>;
using pu = pair<ull, ull>;
 
static constexpr size_t wlen = 1000000;
char work[wlen];
 
vector<vu> conns;
vu plows;
 
ull get_plows(ull cur)
{
	ull sum = 0;
	for (auto c : conns[cur])
	{
		sum += get_plows(c);
	}
	return max(sum, plows[cur]);
}
 
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
 
	ull n;
	cin >> n;
 
	conns.assign(n, {});
	plows.assign(n, 0);
	for (ull i = 0; i < n - 1; ++i)
	{
		ull a, b;
		cin >> a >> b;
		--a, --b;
		conns[a].push_back(b);
	}
	for (ull i = 0; i < n; ++i)
	{
		cin >> plows[i];
	}
 
	cout << get_plows(0) << endl;
}

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