Laskettelukeskus

#include <iostream>
#include <algorithm>

using namespace std;


struct node {
   struct node *parent;
   int snowplows;
   int going = 0;
};



int main()
{
    int slope_count;

    cin >> slope_count;

    node slopes[slope_count];

    node root_node;
    root_node.snowplows = 0;
    slopes[0] = root_node;

    for (int i = 1; i < slope_count; i++) {
        int a, b;
        cin >> a >> b;
        node n;
        n.parent = &slopes[a-1];
        slopes[b-1] = n;
    }

    int plows;
    int slope_i = 0;
    while ( cin >> plows ) {
        slopes[slope_i].snowplows = plows;
        slope_i++;
    }
    for (int i = slope_count; i--;) {
        int needed = max(slopes[i].snowplows, slopes[i].going);
        if (i == 0) {
            cout << needed << endl;
            break;
        }
        slopes[i].parent->going += needed;
    }

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