CSES - Datatähti 2024 alku - Results
Submission details
Task:Laskettelukeskus
Sender:Halfway
Submission time:2023-11-02 12:47:07 +0200
Language:Python3 (CPython3)
Status:READY
Result:0
Feedback
groupverdictscore
#10
#20
Test results
testverdicttimegroup
#1ACCEPTED0.02 s1, 2details
#20.02 s1, 2details
#30.02 s1, 2details
#40.46 s2details
#50.46 s2details
#6ACCEPTED0.02 s1, 2details
#7ACCEPTED0.44 s2details
#8ACCEPTED0.02 s1, 2details
#9ACCEPTED0.36 s2details
#100.02 s1, 2details
#110.36 s2details
#120.02 s1, 2details
#130.36 s2details

Code

n = int(input())
tree = [[int(j) - 1 for j in input().split()] for i in range(n - 1)]
tree.sort(key = lambda k : k[0])
weights = [int(i) for i in input().split()]
maxes = {}
for root, leaf in tree:
    if root in maxes.keys():
        maxes[root] += weights[leaf]
    elif leaf in maxes.keys():
        maxes[root] = max(maxes[leaf], weights[leaf])
        del maxes[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:

input
100
1 73
1 64
64 23
1 88
...

correct output
2675

user output
141

Test 3

Group: 1, 2

Verdict:

input
100
1 36
36 56
56 59
36 97
...

correct output
2808

user output
279

Test 4

Group: 2

Verdict:

input
100000
1 45452
1 74209
45452 78960
45452 79820
...

correct output
28399367694319

user output
5389129540

Test 5

Group: 2

Verdict:

input
100000
1 31165
1 23263
31165 89516
31165 53122
...

correct output
28546840313799

user output
7618244864

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:

input
100
1 2
1 3
2 4
2 5
...

correct output
2809

user output
157

Test 11

Group: 2

Verdict:

input
100000
1 2
1 3
2 4
2 5
...

correct output
26053917212428

user output
1229388190

Test 12

Group: 1, 2

Verdict:

input
100
1 2
1 3
2 4
2 5
...

correct output
5000

user output
200

Test 13

Group: 2

Verdict:

input
100000
1 2
1 3
2 4
2 5
...

correct output
50000000000000

user output
2000000000