Laskettelukeskus

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

#include <iostream>
#include <vector>
using namespace std;

int dfs(const vector<vector<int>>& adj, const vector<int>& machines, int node) {
    if (adj[node].size() == 0) return machines[node];
    int sum = 0;
    for (int i = 0; i < adj[node].size(); i++) {
        sum += dfs(adj, machines, adj[node][i]);
    }
    return max(sum, machines[node]);
}

int main() {
    int n;
    cin >> n;
    vector<vector<int>> adj(n);
    vector<int> machines(n);
    for (int i = 0; i < n - 1; i++) {
        int a;
        int b;
        cin >> a >> b;
        a--;
        b--;
        adj[a].push_back(b);
    }
    for (int i = 0; i < n; i++) cin >> machines[i];
    cout << dfs(adj, machines, 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: 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: WRONG ANSWER

Test 12

Group: 1, 2

Verdict: ACCEPTED

Test 13

Group: 2

Verdict: WRONG ANSWER