Laskettelukeskus

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

#include <bits/stdc++.h>

int main(){
	long long n, o=0;
	int a, b;
	scanf("%lld", &n);
	std::vector<int> r[n];
	long long 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++]);
	//printf("Not input overflow\n");
	std::queue<int> q;
	v[0]=true;
	q.push(0);
	o+=c[0];
	while(!q.empty()){
		a = q.front(); q.pop();
		for(auto u : r[a]){
			if(v[u]) continue;
			v[u] = true;
			if(c[a]-c[u]<0){
				o+=c[u]-c[a];
				c[a]=0;
			}else c[a]-=c[u];
			q.push(u);
		}
	}
	printf("%lld\n", o);
}

Test 1

Group: 1, 2

Verdict: ACCEPTED

Test 2

Group: 1, 2

Verdict: WRONG ANSWER

Test 3

Group: 1, 2

Verdict: WRONG ANSWER

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

Test 11

Group: 2

Verdict: WRONG ANSWER

Test 12

Group: 1, 2

Verdict: ACCEPTED

Test 13

Group: 2

Verdict: ACCEPTED