Laskettelukeskus

#include <bits/stdc++.h>
using namespace std;
 
int main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);
 
    int n;
    cin >> n;
 
    int kerrat[n];
 
    int alaiset[n];
 
    int edelliset[n];
 
    // int yhteydet[n - 1][2];
 
    int alku;
    int loppu;
    for (int i = 0; i < n - 1; i++)
    {
        cin >> alku >> loppu;
        // yhteydet[i][0] = alku - 1;
        // yhteydet[i][1] = loppu - 1;
 
        edelliset[loppu - 1] = alku - 1;
    }
 
    for (int i = 0; i < n; i++)
    {
        cin >> kerrat[i];
        alaiset[i] = 0;
    }
 
    for (int i = n - 1; i > 0; i--)
    {
        // alku = yhteydet[i][0];
        // loppu = yhteydet[i][1];
 
        int nykyinen = i;
        int tarkistettava = edelliset[i];
        int lisays = max(kerrat[i], alaiset[i]);
        while (tarkistettava > i)
        {
            if (lisays == 0)
                break;
 
            if (alaiset[i] < kerrat[nykyinen] && kerrat[nykyinen] < alaiset[i] + lisays)
                lisays = kerrat[i] + lisays - kerrat[nykyinen];
 
            alaiset[tarkistettava] += lisays;
 
            nykyinen = tarkistettava;
            tarkistettava = edelliset[tarkistettava];
        }
 
        alaiset[tarkistettava] += lisays;
    }
 
    cout << max(kerrat[0], alaiset[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: 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