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