Laskettelukeskus

#include <iostream>
#include <vector>

using namespace std;

vector<int> v[101010];
long long d[101010];
long long k[101010];
long long s[101010];
long long n;

void haku(long long s) {
	for (auto u : v[s]) {
		haku(u);
	}
	for (auto i : v[s]) {
		d[s] += d[i];
	}
	if (k[s] > d[s]) d[s] = k[s];
}
int main() {
	cin >> n;
	for (long long i = 1; i <= n - 1; i++) {
		long long a, b;
		cin >> a >> b;
		v[a].push_back(b);
	}
	long long eka;
	cin >> eka;
	k[1] = eka;
	for (long long i = 2; i <= n; i++) {
		long long c;
		cin >> c;
		k[i] = c;
	}
	haku(1);
	cout << d[1];
}

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