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