Laskettelukeskus

input/code.cpp: In function 'int main()':
input/code.cpp:25:20: warning: unused variable 'ans' [-Wunused-variable]
   25 |     ll n, i, a, b, ans = 0;;
      |                    ^~~

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef vector <ll> vl;
typedef vector <bool> vb;
typedef vector <vl> graph;

ll dfs(const graph &g, vb &v, const vl &need, ll cur)
{
    ll k = 0;
    for(auto x : g[cur])
        if(!v[x])
        {
            v[x] = true;
            k += dfs(g, v, need, x);
        }
    return max(k, need[cur]);
}

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

    ll n, i, a, b, ans = 0;;
    cin >> n;
    graph g(n + 1); vb v(n + 1, false); vl need(n + 1);
    v[1] = true;
    for(i = 1; i < n; i++)
    {
        cin >> a >> b;
        g[a].push_back(b);
    }
    for(i = 1; i <= n; i++)
        cin >> need[i];

    cout << dfs(g, v, need, 1);

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