CSES - Datatähti 2024 alku - Results
Submission details
Task:Laskettelukeskus
Sender:Olive
Submission time:2023-11-01 21:07:48 +0200
Language:Python3 (PyPy3)
Status:READY
Result:0
Feedback
groupverdictscore
#10
#20
Test results
testverdicttimegroup
#10.04 s1, 2details
#20.05 s1, 2details
#30.05 s1, 2details
#4--2details
#5--2details
#6ACCEPTED0.05 s1, 2details
#7--2details
#80.05 s1, 2details
#9--2details
#100.05 s1, 2details
#11--2details
#120.05 s1, 2details
#13--2details

Code

n = int(input())

slopes = {}
for i in range(n-1):
    slopes[i] = [int(x) for x in input().split()]

snowamount = [int(x) for x in input().split()]

amount = 0

def find_pair(num, daddy_number):
    global amount
    pairs = []
    

    
    for i in range(n-1):
        if(slopes[i][0] == num):
            pairs.append(slopes[i][1])
    number = 0
    if(len(pairs) > 0):
        for i in range(len(pairs)):
            number += snowamount[pairs[i]-1]
            find_pair(pairs[i], snowamount[pairs[i]-1])
    if(number < daddy_number):
        amount = amount + (daddy_number - number)
            
    if(len(pairs) == 0):
        amount = amount + daddy_number


for i in range(n-1):
    if(slopes[i][0] == 1):
        num = slopes[i][1]
        find_pair(slopes[i][1], snowamount[0])
      


print(amount)

Test details

Test 1

Group: 1, 2

Verdict:

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

correct output
6

user output
8

Test 2

Group: 1, 2

Verdict:

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

correct output
2675

user output
5040

Test 3

Group: 1, 2

Verdict:

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

correct output
2808

user output
5666

Test 4

Group: 2

Verdict:

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

correct output
28399367694319

user output
(empty)

Test 5

Group: 2

Verdict:

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

correct output
28546840313799

user output
(empty)

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:

input
100000
1 66038
1 56789
56789 7403
66038 69542
...

correct output
0

user output
(empty)

Test 8

Group: 1, 2

Verdict:

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

correct output
100

user output
200

Test 9

Group: 2

Verdict:

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

correct output
1000000000

user output
(empty)

Test 10

Group: 1, 2

Verdict:

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

correct output
2809

user output
5608

Test 11

Group: 2

Verdict:

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

correct output
26053917212428

user output
(empty)

Test 12

Group: 1, 2

Verdict:

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

correct output
5000

user output
10000

Test 13

Group: 2

Verdict:

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

correct output
50000000000000

user output
(empty)