Laskettelukeskus

#include <iostream>

using namespace std;

// Create the tree. Traverse tree from bottom
// First create all the nodes in vector
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;

        //cout << a << ", " << b << endl;
    }

    int plows;
    int slope_i = 0;
    while ( cin >> plows ) {
        slopes[slope_i].snowplows = plows;
        //cout << "Hill " << slope_i << " needs: " << plows << endl;
        slope_i++;
    }
    //cout << "Best route code" << endl;
    for (int i = slope_count; i--;) {
        int difference = slopes[i].snowplows - slopes[i].going;
        if (difference < 0) {
            difference = 0;
        }

        if (slopes[i].going == slopes[i].snowplows) {
            difference = slopes[i].going;
        }

        if (i == 0) {
            //cout << "END " << slopes[i].going + difference << endl;
            cout << slopes[i].going + difference << endl;
            break;
        }
        //cout << "Before " << slopes[i].parent->going << endl;
        slopes[i].parent->going += difference;
        //cout << "After " << slopes[i].parent->going << endl;
        //cout << "After " << slopes[i].parent << endl;
        //cout << "index " << i << " --- Plows " << slopes[i].snowplows << " going trough " << slopes[i].going << " diff " << difference << " par " << slopes[i].parent->going << endl;
    }

        //cout << slope_count << endl;
        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: 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