Laskettelukeskus

input/code.cpp: In function 'int main()':
input/code.cpp:30:10: warning: unused variable 'input_char' [-Wunused-variable]
   30 |     char input_char[n];
      |          ^~~~~~~~~~

#include <bits/stdc++.h>

# define NMAX 100001

std::vector<uint64_t> adjacent_v[NMAX];
uint64_t n;
uint64_t weights[NMAX];
bool visited[NMAX];


uint64_t dfs(uint64_t node) {

    uint64_t max_val = 0;

    visited[node] = true;
    for (auto u: adjacent_v[node]) {
        if(visited[u] == true){
            continue;
        } else
        {
            max_val += dfs(u);
        }         
    }
    return std::max(max_val, weights[node]);
}

int main() {

    std::cin >> n;
    char input_char[n];

    uint64_t a, b;

    for (size_t i = 0; i < n - 1; i++)
    {
        std::cin >> a >> b;
        adjacent_v[a-1].push_back(b-1);
    }

    for (size_t i = 0; i < n; i++)
    {
        std::cin >> weights[i];
    }
    

    for (size_t i = 0; i < n; i++)
    {
        visited[i] = false;
    }


    uint64_t max_val = dfs(0);
    
    std::cout << max_val << std::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: 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