Laskettelukeskus

input/code.cpp: In function 'long long int f(int)':
input/code.cpp:30:27: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   30 |         for (int i = 0; i < c[z].size(); i++) x += f(c[z][i]);
      |                         ~~^~~~~~~~~~~~~

#include <iostream>
#include <vector>

using namespace std;

long long *v;
vector<int> *c;
long long f(int z);

int main() {
	int n;
	cin >> n;
	long long a[n];
	vector<int> b[n];
	for (int i = 0; i < n-1; i++) {
		int s, t;
		cin >> s >> t;
		b[s-1].push_back(t-1);
	}
	for (int i = 0; i < n; i++) cin >> a[i];
	v = a, c = b;
	long long r = f(0);
	cout << r << "\n";

	return 0;
}

long long f(int z) {
	long long x = 0;
	for (int i = 0; i < c[z].size(); i++) x += f(c[z][i]);
	if (v[z] > x) x = v[z];
	return x;
}

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