Laskettelukeskus

#include <iostream>
#include <vector>
using namespace std;

typedef long long ll;

vector<vector<int>> g(100001);
int c[100001];
ll res[100001];

void deefes(int s, int e)
{
    for (auto it : g[s])
    {
        if (it == e) continue;
        
        deefes(it, s);
        res[s] += res[it];
    }

    res[s] = max(res[s], (ll)c[s]);
}

int main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);

    int n;
    cin >> n;

    for (int z = 0; z < n - 1; z++)
    {
        int u;
        int v;

        cin >> u >> v;

        g[u].push_back(v);
        g[v].push_back(u);
    }

    for (int zz = 1; zz <= n; zz++)
    {
        cin >> c[zz];
    }

    deefes(1,0);

    cout << res[1];
}

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

Test 5

Group: 2

Verdict: ACCEPTED

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