Laskettelukeskus

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

#include <iostream>
#include <vector>

#define PER long int

struct Rinne
{
    PER amount;
    std::vector<int> rinteet;
};


PER solve(const std::vector<Rinne>& rinteet, int rinne) {
    PER sum = 0;
    for (int i=0; i < rinteet[rinne].rinteet.size(); i++) {
        sum += solve(rinteet, rinteet[rinne].rinteet[i]);
    }
    if (rinteet[rinne].amount > sum) { return rinteet[rinne].amount; } 
    else { return sum; }
}

Rinne new_node() {
    std::vector<int> _a = {};
    Rinne _r;
    _r.amount = 0;
    _r.rinteet = _a;
    return _r;
}


int main() {
    std::vector<Rinne> rinteet;
    rinteet.push_back(new_node());

    int amount;
    std::cin >> amount;

    int a, b;
    for (int i = 0; i < amount-1; i++) {
        std::cin >> a >> b;
        rinteet.push_back(new_node());
        rinteet[a-1].rinteet.push_back(b-1);
    }
    for (int i = 0; i < amount; i++) {
        std::cin >> a;
        rinteet[i].amount = a;
    }

    std::cout << solve(rinteet, 0);

    return 0;
}

Test 1

Group: 1, 2

Verdict: ACCEPTED

Test 2

Group: 1, 2

Verdict: RUNTIME ERROR

Test 3

Group: 1, 2

Verdict: RUNTIME ERROR

malloc(): corrupted top size

Test 4

Group: 2

Verdict: RUNTIME ERROR

Test 5

Group: 2

Verdict: RUNTIME ERROR

Test 6

Group: 1, 2

Verdict: RUNTIME ERROR

terminate called after throwing an instance of 'std::bad_alloc'
  what():  std::bad_alloc

Test 7

Group: 2

Verdict: RUNTIME ERROR

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