Laskettelukeskus

input/code.cpp:61:1: warning: "/*" within comment [-Wcomment]
   61 | /*
      |  
input/code.cpp: In function 'int main()':
input/code.cpp:31:8: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   31 |   scanf("%d\n", &count);
      |   ~~~~~^~~~~~~~~~~~~~~~
input/code.cpp:37:10: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   37 |     scanf("%u %u\n", &temp1, &temp2);
      |     ~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~
input/code.cpp:50:10: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   50 |     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<unsigned int, std::vector<unsigned int>> nodes;
// nodes => hash? reduce mem usage

int calculate(int cur_node) {
  std::vector<unsigned int> children = nodes[cur_node];
  uint32_t t = 0;
  for (int 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);
  int count;
  //std::cin >> count;
  scanf("%d\n", &count);

  unsigned int temp1;
  unsigned int temp2;

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


  //std::string input;
  //std::getline(std::cin, input);
  //std::getline(std::cin, input);
  //std::cin.ignore();
  //std::istringstream ss(input);

  for (int i = 0; i < count; i++) {
    scanf("%u", &requireds[i]);
    //std::cin >> requireds[i];
  }
  /*
  std::cout << "[";
  for (auto chara : input) {
      std::cout << chara;
  }
  
  std::cout << "]";

/*   
  for (int i = 0; i < count; i++) {
    std::cin >> 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