CSES - HIIT Open 2024 - Results
Submission details
Task:Key cutting
Sender:Tunkkari
Submission time:2024-11-16 14:26:09 +0200
Language:Python3 (PyPy3)
Status:READY
Result:ACCEPTED
Test results
testverdicttime
#1ACCEPTED0.04 sdetails
#2ACCEPTED0.04 sdetails
#3ACCEPTED0.04 sdetails
#4ACCEPTED0.04 sdetails
#5ACCEPTED0.04 sdetails
#6ACCEPTED0.04 sdetails
#7ACCEPTED0.04 sdetails
#8ACCEPTED0.04 sdetails
#9ACCEPTED0.07 sdetails
#10ACCEPTED0.08 sdetails
#11ACCEPTED0.13 sdetails
#12ACCEPTED0.15 sdetails
#13ACCEPTED0.19 sdetails
#14ACCEPTED0.18 sdetails

Code

n = int(input())

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



def inner(start,end):
    if start == end:

        total = 1
        if segments[start] == 0:
            total = 0
    else:
        min_value = min(segments[start:end+1])
        idxs = []
        for i in range(start,end+1):
            if segments[i] == min_value:
                idxs.append(i)
            elif segments[i] < min_value:
                idxs = []
                min_value = segments[i]

        prev = idxs[0]
        total = 0
        if (min_value > 0):
            total = 1

        if prev > start:
            total += inner(start,prev-1)


        for k in idxs[1:]:
            if prev != k -1:            
                new_start = prev+1
                new_end = k -1
                total += inner(new_start,new_end)
            prev = k

        if end > prev:
            total += inner(prev+1,end)
        
    return total


print(inner(0,n-1))

Test details

Test 1

Verdict: ACCEPTED

input
3
1 2 1

correct output
2

user output
2

Test 2

Verdict: ACCEPTED

input
1
0

correct output
0

user output
0

Test 3

Verdict: ACCEPTED

input
1
9

correct output
1

user output
1

Test 4

Verdict: ACCEPTED

input
100
0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 ...

correct output
0

user output
0

Test 5

Verdict: ACCEPTED

input
100
0 0 0 1 1 0 0 0 0 1 1 1 0 1 1 ...

correct output
25

user output
25

Test 6

Verdict: ACCEPTED

input
100
2 1 2 1 2 0 0 0 1 1 2 2 1 2 2 ...

correct output
41

user output
41

Test 7

Verdict: ACCEPTED

input
100
36 5 10 37 94 59 20 31 64 2 58...

correct output
99

user output
99

Test 8

Verdict: ACCEPTED

input
100
228768416 32415139 952687252 6...

correct output
100

user output
100

Test 9

Verdict: ACCEPTED

input
100000
0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 ...

correct output
0

user output
0

Test 10

Verdict: ACCEPTED

input
100000
1 1 1 0 0 1 0 0 0 0 1 0 1 0 0 ...

correct output
24965

user output
24965

Test 11

Verdict: ACCEPTED

input
100000
2 1 2 2 2 2 2 1 1 0 1 1 0 1 1 ...

correct output
38968

user output
38968

Test 12

Verdict: ACCEPTED

input
100000
4 4 5 4 4 5 0 2 2 1 4 4 1 0 5 ...

correct output
59156

user output
59156

Test 13

Verdict: ACCEPTED

input
100000
18 5 6 16 8 10 1 7 4 15 5 9 19...

correct output
82598

user output
82598

Test 14

Verdict: ACCEPTED

input
100000
33 37 37 86 42 38 18 10 77 57 ...

correct output
94897

user output
94897