| Task: | Laskettelukeskus |
| Sender: | Halfway |
| Submission time: | 2023-11-03 09:34:11 +0200 |
| Language: | Python3 (CPython3) |
| Status: | READY |
| Result: | 100 |
| group | verdict | score |
|---|---|---|
| #1 | ACCEPTED | 53 |
| #2 | ACCEPTED | 47 |
| test | verdict | time | group | |
|---|---|---|---|---|
| #1 | ACCEPTED | 0.02 s | 1, 2 | details |
| #2 | ACCEPTED | 0.02 s | 1, 2 | details |
| #3 | ACCEPTED | 0.02 s | 1, 2 | details |
| #4 | ACCEPTED | 0.42 s | 2 | details |
| #5 | ACCEPTED | 0.42 s | 2 | details |
| #6 | ACCEPTED | 0.02 s | 1, 2 | details |
| #7 | ACCEPTED | 0.38 s | 2 | details |
| #8 | ACCEPTED | 0.02 s | 1, 2 | details |
| #9 | ACCEPTED | 0.38 s | 2 | details |
| #10 | ACCEPTED | 0.02 s | 1, 2 | details |
| #11 | ACCEPTED | 0.37 s | 2 | details |
| #12 | ACCEPTED | 0.02 s | 1, 2 | details |
| #13 | ACCEPTED | 0.38 s | 2 | details |
Code
n = int(input())
tree = [[int(j) - 1 for j in input().split()] for i in range(n - 1)][::-1]
weights = [int(i) for i in input().split()]
maxes = {}
for root, leaf in tree:
if root in maxes.keys():
try:
maxes[root] += max(maxes[leaf], weights[leaf])
except:
maxes[root] += weights[leaf]
elif leaf in maxes.keys():
maxes[root] = max(maxes[leaf], weights[leaf])
else:
maxes[root] = weights[leaf]
print(max(maxes[0], weights[0]))Test details
Test 1
Group: 1, 2
Verdict: ACCEPTED
| input |
|---|
| 5 1 2 1 3 3 4 3 5 ... |
| correct output |
|---|
| 6 |
| user output |
|---|
| 6 |
Test 2
Group: 1, 2
Verdict: ACCEPTED
| input |
|---|
| 100 1 73 1 64 64 23 1 88 ... |
| correct output |
|---|
| 2675 |
| user output |
|---|
| 2675 |
Test 3
Group: 1, 2
Verdict: ACCEPTED
| input |
|---|
| 100 1 36 36 56 56 59 36 97 ... |
| correct output |
|---|
| 2808 |
| user output |
|---|
| 2808 |
Test 4
Group: 2
Verdict: ACCEPTED
| input |
|---|
| 100000 1 45452 1 74209 45452 78960 45452 79820 ... |
| correct output |
|---|
| 28399367694319 |
| user output |
|---|
| 28399367694319 |
Test 5
Group: 2
Verdict: ACCEPTED
| input |
|---|
| 100000 1 31165 1 23263 31165 89516 31165 53122 ... |
| correct output |
|---|
| 28546840313799 |
| user output |
|---|
| 28546840313799 |
Test 6
Group: 1, 2
Verdict: ACCEPTED
| input |
|---|
| 100 1 79 79 9 79 45 45 10 ... |
| correct output |
|---|
| 0 |
| user output |
|---|
| 0 |
Test 7
Group: 2
Verdict: ACCEPTED
| input |
|---|
| 100000 1 66038 1 56789 56789 7403 66038 69542 ... |
| correct output |
|---|
| 0 |
| user output |
|---|
| 0 |
Test 8
Group: 1, 2
Verdict: ACCEPTED
| input |
|---|
| 100 1 2 2 3 3 4 4 5 ... |
| correct output |
|---|
| 100 |
| user output |
|---|
| 100 |
Test 9
Group: 2
Verdict: ACCEPTED
| input |
|---|
| 100000 1 2 2 3 3 4 4 5 ... |
| correct output |
|---|
| 1000000000 |
| user output |
|---|
| 1000000000 |
Test 10
Group: 1, 2
Verdict: ACCEPTED
| input |
|---|
| 100 1 2 1 3 2 4 2 5 ... |
| correct output |
|---|
| 2809 |
| user output |
|---|
| 2809 |
Test 11
Group: 2
Verdict: ACCEPTED
| input |
|---|
| 100000 1 2 1 3 2 4 2 5 ... |
| correct output |
|---|
| 26053917212428 |
| user output |
|---|
| 26053917212428 |
Test 12
Group: 1, 2
Verdict: ACCEPTED
| input |
|---|
| 100 1 2 1 3 2 4 2 5 ... |
| correct output |
|---|
| 5000 |
| user output |
|---|
| 5000 |
Test 13
Group: 2
Verdict: ACCEPTED
| input |
|---|
| 100000 1 2 1 3 2 4 2 5 ... |
| correct output |
|---|
| 50000000000000 |
| user output |
|---|
| 50000000000000 |