Laskettelukeskus

#include <iostream>
 
using namespace std;
 
int t = 0;
 
struct rin
{
    rin* up = nullptr;
    int left = 0;
    int nleft = 0;
    bool down = false;
 
    void upd(int x)
    {
        if(!up)
            t += x;
 
        left -= x;
        if (up)
            up->upd(x);
    }
 
    void upd2(int x)
    {
        if (!up)
            t += x;
        else
        {
            nleft = x;
 
            int dx = x;
 
            if (up->nleft > 0)
            {
                dx = min(0, x - up->nleft);
                up->nleft = max(0, up->nleft - x);
            }
 
            left -= x;
            if (dx)
                up->upd2(dx);
        }
    }
};
 
int main()
{
    rin rinL[1002];
    
    int n;
    cin >> n;
 
    for (int i = 0; i < n - 1; i++)
    {
        int a, b;
        cin >> a >> b;
        rinL[b].up = &rinL[a];
        rinL[a].down = true;
    }
 
    for (int i = 1; i < n + 1; i++)
    {
        int a;
        cin >> a;
        rinL[i].left = a;
    }
 
    for (int i = 1; i < n + 1; i++)
    {
        auto& r = rinL[i];
        if (r.down == false)
            r.upd(r.left);
    }
 
    for (int i = 1; i < n + 1; i++)
    {
        auto& r = rinL[i];
        if (r.left > 0)
            r.upd(r.left);
    }
 
    cout << t;
 
    return 0;
}

Test 1

Group: 1, 2

Verdict: ACCEPTED

Test 2

Group: 1, 2

Verdict: WRONG ANSWER

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: WRONG ANSWER

Test 11

Group: 2

Verdict: RUNTIME ERROR

Test 12

Group: 1, 2

Verdict: ACCEPTED

Test 13

Group: 2

Verdict: RUNTIME ERROR