Laskettelukeskus

#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define N 112345

int n,a,b;
ll r,ap[N],c[N];
vector<int> v[N];

void haku(int s,int p){
	ll S=0;
	for(int u:v[s]){
		if(u==p) continue;
		haku(u,s);
		S+=ap[u];
	}
	ap[s]=max(S,c[s]);
}

int main(){
	cin.tie(0)->sync_with_stdio(0);
	cin>>n;
	for(int i=0;i<n-1;++i){
		cin>>a>>b;
		v[a].push_back(b);
		v[b].push_back(a);
	}
	for(int i=0;i<n;++i) cin>>c[i+1];
	haku(1,1);
	cout<<ap[1]<<'\n';
}

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