Laskettelukeskus

#include <iostream>
#include <vector>

#define PER long int

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


PER solve(const Rinne* rinteet, int rinne) {
    PER sum = 0;
    for (int i : rinteet[rinne].rinteet) {
        sum += solve(rinteet, i);
    }
    return (rinteet[rinne].amount > sum ? rinteet[rinne].amount : sum);
}


int main() {
    Rinne rinteet[1000];
    rinteet[0] = {0, {}};

    int amount;
    std::cin >> amount;

    int a, b;
    for (int i = 1; i < amount; i++) {
        std::cin >> a >> b;
        // rinteet[i] = {0, {}};
        // rinteet[a-1].rinteet.push_back(b-1);
    }
    for (int i = 0; i < amount; i++) {
        std::cin >> a;
        // rinteet[i].amount = a;
    }
    
    rinteet[0] = {0, {1, 2}};
    rinteet[1] = {2, {}};
    rinteet[2] = {4, {3, 4}};
    rinteet[3] = {3, {}};
    rinteet[4] = {1, {}};
    std::cout << solve(rinteet, 0);
    return 0;
}

Test 1

Group: 1, 2

Verdict: ACCEPTED

Test 2

Group: 1, 2

Verdict: WRONG ANSWER

Test 3

Group: 1, 2

Verdict: WRONG ANSWER

Test 4

Group: 2

Verdict: WRONG ANSWER

Test 5

Group: 2

Verdict: WRONG ANSWER

Test 6

Group: 1, 2

Verdict: WRONG ANSWER

Test 7

Group: 2

Verdict: WRONG ANSWER

Test 8

Group: 1, 2

Verdict: WRONG ANSWER

Test 9

Group: 2

Verdict: WRONG ANSWER

Test 10

Group: 1, 2

Verdict: WRONG ANSWER

Test 11

Group: 2

Verdict: WRONG ANSWER

Test 12

Group: 1, 2

Verdict: WRONG ANSWER

Test 13

Group: 2

Verdict: WRONG ANSWER