Laskettelukeskus

input/code.cpp: In function 'int main()':
input/code.cpp:30:8: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   30 |   scanf("%d\n", &count);
      |   ~~~~~^~~~~~~~~~~~~~~~
input/code.cpp:36:10: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   36 |     scanf("%u %u\n", &temp1, &temp2);
      |     ~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~
input/code.cpp:41:10: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   41 |     scanf("%u", &requireds[i]);
      |     ~~~~~^~~~~~~~~~~~~~~~~~~~~

#include <string>
#include <array>
#include <iostream>
#include <sstream>
#include <vector>
#include <ios>
#include <cstdio>
#include <unordered_map>

std::array<uint32_t, 100000> requireds;
std::unordered_map<uint32_t, std::vector<uint32_t>> nodes;
// nodes => hash? reduce mem usage

uint32_t calculate(uint32_t cur_node) {
  std::vector<uint32_t> children = nodes[cur_node];
  uint64_t t = 0;
  for (uint32_t c : children) {
    t += calculate(c);
  }
  if (t > requireds[cur_node - 1]) {
    return t;
  } else {
    return requireds[cur_node - 1];
  }
}

int main() {
  std::ios_base::sync_with_stdio(false);
  uint32_t count;
  scanf("%d\n", &count);

  uint32_t temp1;
  uint32_t temp2;

  for (uint32_t i = 0; i < (count - 1); i++) {
    scanf("%u %u\n", &temp1, &temp2);
    nodes[temp1].push_back(temp2);
  }

  for (uint32_t i = 0; i < count; i++) {
    scanf("%u", &requireds[i]);
  }

  std::cout << calculate(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: 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