Laskettelukeskus

#include <iostream>
#include <vector>

using namespace std;

vector<int> adj[100005];
int nodeSnow[100005] = {};

int dfs(int s) {
    int parentSnow = nodeSnow[s];
    int childSnow = 0;
    for (auto u : adj[s]) {
        childSnow += dfs(u);
    }
    return max(childSnow, parentSnow);
}

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

    for (int i = 0; i < n - 1; i++) {
        int a, b;
        cin >> a >> b;
        a--;
        b--;
        adj[a].push_back(b);
    }

    for (int i = 0; i < n; i++) {
        int a;
        cin >> a;
        nodeSnow[i] = a;
    }

    cout << dfs(0) << 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