Laskettelukeskus

input/code.cpp: In function 'int main()':
input/code.cpp:19:14: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   19 |         scanf("%d", &n);
      |         ~~~~~^~~~~~~~~~
input/code.cpp:24:22: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   24 |                 scanf("%d %d", &a, &b);
      |                 ~~~~~^~~~~~~~~~~~~~~~~
input/code.cpp:28:25: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   28 |         while(i<n) scanf("%lld", &c[i++]);
      |                    ~~~~~^~~~~~~~~~~~~~~~~

// depth first
#include <bits/stdc++.h>

typedef long long ll;

using namespace std;

vector<int> *rr;
ll *cc;

ll dfs(int s, int e){
	ll o = 0;
	for(auto u : rr[s]) if(u != e) o+=dfs(u, s);
	return max(cc[s], o);	
}

int main(){
	int a, b, n;
	scanf("%d", &n);
	vector<int> r[n];
	ll c[n], i=n-1;
	//std::vector<bool> v(n, false);
	while(i--){
		scanf("%d %d", &a, &b);
		r[a-1].push_back(b-1);
		r[b-1].push_back(a-1);
	}i=0;
	while(i<n) scanf("%lld", &c[i++]);
	cc = &c[0];
	rr = &r[0];

	printf("%lld\n", dfs(0, 0));
	//printf("Not input overflow\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