Laskettelukeskus

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

#include <bits/stdc++.h>

using namespace std;

vector<long long> v[100001];
vector<long long> s;
long long smax;

long long search(long long z) {
    long long sum;
    sum = 0;
    for (int i = 0; i < v[z].size(); i++) {
        sum = sum+search(v[z][i]);
    }
    sum = max(sum,s[z-1]);
    return sum;
}

int main() {
    smax = -1;
    long long n,a,b;
    cin >> n;
    for (int i = 1; i < n; i++) {
        cin >> a >> b;
        v[a].push_back(b);
    }
    for (int i = 0; i < n; i++) {
        cin >> a;
        s.push_back(a);
    }
    cout << search(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