Laskettelukeskus

#include <bits/stdc++.h>

using namespace std;

class Slope
{
public:
    int id;
    int plowsNeeded;
    vector<int> children;

    Slope(int _id, int _plowsNeeded) : id(_id), plowsNeeded(_plowsNeeded) {}
};

int minSnowplows(vector<Slope> &slopes, int currentSlope)
{
    int totalPlows = 0;

    for (int child : slopes[currentSlope].children)
    {
        totalPlows += minSnowplows(slopes, child);
    }

    int additionalPlows = max(0, slopes[currentSlope].plowsNeeded - totalPlows);
    totalPlows += additionalPlows;

    return totalPlows;
}

int main()
{
    int n;
    cin >> n;

    vector<Slope> slopes(n + 1, Slope(0, 0)); 

    for (int i = 1; i < n; ++i)
    {
        int a, b;
        cin >> a >> b;
        slopes[a].children.push_back(b);
        slopes[b].id = b;
    }

    for (int i = 1; i <= n; ++i)
    {
        cin >> slopes[i].plowsNeeded;
    }

    int result = minSnowplows(slopes, 1); 
    cout << result << endl;

    return 0;
}

Test 1

Group: 1, 2

Verdict: ACCEPTED

Test 2

Group: 1, 2

Verdict: ACCEPTED

Test 3

Group: 1, 2

Verdict: ACCEPTED

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