Laskettelukeskus

#include <bits/stdc++.h>

using namespace std;

vector<int> reitit[100001];
int auraukset[100001];

int aurauksia(int alku){
  int value = auraukset[alku];
  int lapset = 0;

  for(auto lapsi: reitit[alku]){
    lapset += aurauksia(lapsi);
  }

  return max(lapset, value);
}

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

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

  //Auraukset
  for(int i = 0; i<n; i++){
    cin >> auraukset[i+1];
  }

  cout << aurauksia(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: 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