Laskettelukeskus

#include <iostream>
#include <vector>

using namespace std;


struct rin
{
    vector<rin*> down;
    int l = 0;

    int upd()
    {
        int t = 0;
        for (auto r : down)
            t += r->upd();

        return max(t, l);
    }
};

int main()
{
    rin rinL[1002];
    
    int n;
    cin >> n;

    for (int i = 0; i < n - 1; i++)
    {
        int a, b;
        cin >> a >> b;
        rinL[a].down.push_back(& rinL[b]);
    }

    for (int i = 1; i < n + 1; i++)
    {
        int a;
        cin >> a;

        rinL[i].l = a;
    }

    cout << rinL[1].upd();

    return 0;
}

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: RUNTIME ERROR

Test 5

Group: 2

Verdict: RUNTIME ERROR

Test 6

Group: 1, 2

Verdict: ACCEPTED

Test 7

Group: 2

Verdict: RUNTIME ERROR

Test 8

Group: 1, 2

Verdict: ACCEPTED

Test 9

Group: 2

Verdict: RUNTIME ERROR

Test 10

Group: 1, 2

Verdict: ACCEPTED

Test 11

Group: 2

Verdict: RUNTIME ERROR

Test 12

Group: 1, 2

Verdict: ACCEPTED

Test 13

Group: 2

Verdict: RUNTIME ERROR