Laskettelukeskus

#include <iostream>
#include <vector>
#include <sstream>

struct Node {
    Node* left            = nullptr;
    Node* right           = nullptr;
    size_t ploughs_needed;

    size_t total_ploughs() const {
        size_t l = 0, r = 0;
        if (left) l = left->total_ploughs();
        if (right) r = right->total_ploughs();

        return std::max(l + r, ploughs_needed);
    }
};

int main() {
//    auto s = std::string(R"(5
//1 2
//1 3
//3 4
//3 5
//0 2 4 3 1)");
//    std::istringstream input(s);
        auto& input = std::cin;

    size_t count;
    input >> count;
    std::vector<Node> nodes(count);

    for (size_t i = 0; i < count - 1; ++i) {
        size_t id1, id2;
        input >> id1 >> id2;
        --id1;
        --id2;
        // id1 is higher than id2, so a plough can go id1 --> id2
        if (nodes[id1].left == nullptr) {
            nodes[id1].left = &nodes[id2];
        } else {
            nodes[id1].right = &nodes[id2];
        }
    }

    for (size_t i = 0; i < count; ++i) {
        size_t num;
        input >> num;
        nodes[i].ploughs_needed = num;
    }

    std::cout << nodes[0].total_ploughs() << "\n";
}

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

Test 12

Group: 1, 2

Verdict: ACCEPTED

Test 13

Group: 2

Verdict: ACCEPTED