Laskettelukeskus

#include <iostream>
#include <vector>

using namespace std;

vector<pair<long long,long long>> v[101010];
vector<pair<long long, long long>> w;
long long z[101010];
long long d[101010];
long long k[101010];
long long s[101010];
long long n;

void haku(long long s) {
	if (z[s]) return;
	z[s] = 1;
	for (auto u : v[s]) {
		haku(u.first);
	}
	for (auto i : v[s]) {
		d[s] += d[i.first];
	}
	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;
		w.push_back({ a, b });
	}
	long long eka;
	cin >> eka;
	k[1] = eka;
	for (long long i = 1; i <= n-1; i++) {
		long long c;
		cin >> c;
		v[w[i-1].first].push_back({ w[i-1].second, c });
		k[w[i-1].second] = c;
	}
	haku(1);
	cout << d[1];
}

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: ACCEPTED

Test 11

Group: 2

Verdict: ACCEPTED

Test 12

Group: 1, 2

Verdict: ACCEPTED

Test 13

Group: 2

Verdict: ACCEPTED