Laskettelukeskus

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
ll n, q, m, a, b, c;
vector<pair<int,int>> v[202020];
int e[202020];
int d[202020];
int p[202020];
pair<int,int> r[202020];
set<int> se;
string s;

void haku(int s, int syv) {
	d[s]=syv;
//	cout << syv << "\n";
	for (auto u: v[s]) {
		haku(u.first,syv+1);
	}
}


int main() {
	ios_base::sync_with_stdio(0);
	cin.tie(0);
	cin >> n;

	a=5; b=6;
	for (int i=2; i<=n; i++) {
		cin >> a >> b;
		r[i]={a,b};
        }
	cin >> p[1];
	int aurat=p[1];
	for (int i=2; i<=n; i++) {
		cin >> p[i];
		v[r[i].first].push_back({r[i].second,p[i]});
		p[i]=0;
	}
	p[1]=0; p[0]=0;
	haku(1,0);
	for (int i=1; i<=n; i++) {
		for (auto u: v[i]) {
			p[d[i]]+=u.second;
		}
	}
	for (int i=0; i<=n; i++) {
		aurat=max(aurat,p[i]);
	}
	cout << aurat << "\n";
}

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

Test 13

Group: 2

Verdict: WRONG ANSWER