CSES - Datatähti 2021 alku - Results
Submission details
Task:Alitaulukot
Sender:Kaapipo
Submission time:2020-09-29 10:38:42 +0300
Language:Python3 (CPython3)
Status:READY
Result:0
Feedback
groupverdictscore
#10
#20
#30
Test results
testverdicttimegroup
#1ACCEPTED0.03 s1, 2, 3details
#20.03 s1, 2, 3details
#3ACCEPTED0.03 s1, 2, 3details
#40.03 s1, 2, 3details
#5ACCEPTED0.03 s1, 2, 3details
#6--2, 3details
#70.04 s2, 3details
#80.04 s2, 3details
#9--2, 3details
#10--2, 3details
#11--3details
#120.38 s3details
#130.38 s3details
#14--3details
#15--3details
#16--3details
#17--3details

Code

from functools import reduce
# Syöte:
# 5 2
# 3 1 2 6 4
#
# Tuloste:
# 9
#
# Selitys: Alitaulukot ovat [3], [1], [2], [6], [4], [3,1], [1,2], [6,4] ja [3,1,2].

n, k = map(int, input().split(" "))


def sub_lists (L):
    return [L[i:i+j] for i in range(0,len(L)) for j in range(1,len(L)-i+1)]

# n, k = 5, 2
array = list(map(int, input().split(" ")))

p1 = 0
p2 = 1

slices = []

lastSlice = []

while p2 <= len(array):
    slice = array[p1:p2]

    mi = min(slice)
    ma = max(slice)

    if ma - mi > k:
        p1 += 1
        if lastSlice:
            slices.append(lastSlice)
            lastSlice = []
    else:
        p2 += 1
        # print(slice)
        lastSlice = slice
slices.append(lastSlice)
# print(slices)
s = map(sub_lists, slices)
# print(*s)
print(sum(map(len, s)))
# s = map(lambda x: x * (x + 1) // 2, map(len, slices))
# print(sum(list(s)))

Test details

Test 1

Group: 1, 2, 3

Verdict: ACCEPTED

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

correct output
5050

user output
5050

Test 2

Group: 1, 2, 3

Verdict:

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

correct output
317

user output
346

Test 3

Group: 1, 2, 3

Verdict: ACCEPTED

input
100 10
71 60 61 96 25 10 10 9 84 85 1...

correct output
119

user output
119

Test 4

Group: 1, 2, 3

Verdict:

input
100 990000000
111122929 961821360 578238211 ...

correct output
4006

user output
5812

Test 5

Group: 1, 2, 3

Verdict: ACCEPTED

input
100 1000000000
553190572 453407680 667300705 ...

correct output
5050

user output
5050

Test 6

Group: 2, 3

Verdict:

input
2000 0
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
2001000

user output
(empty)

Test 7

Group: 2, 3

Verdict:

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

correct output
6340

user output
6960

Test 8

Group: 2, 3

Verdict:

input
2000 10
65 88 33 88 41 10 17 38 22 3 8...

correct output
2413

user output
2432

Test 9

Group: 2, 3

Verdict:

input
2000 999000000
746120950 772769620 721488968 ...

correct output
1287776

user output
(empty)

Test 10

Group: 2, 3

Verdict:

input
2000 1000000000
621947980 510355354 756705418 ...

correct output
2001000

user output
(empty)

Test 11

Group: 3

Verdict:

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

correct output
5000050000

user output
(empty)

Test 12

Group: 3

Verdict:

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

correct output
317066

user output
349938

Test 13

Group: 3

Verdict:

input
100000 10
10 3 6 3 43 60 5 48 15 27 86 4...

correct output
123292

user output
124512

Test 14

Group: 3

Verdict:

input
100000 999990000
460235639 963048588 47270983 3...

correct output
4946886742

user output
(empty)

Test 15

Group: 3

Verdict:

input
100000 1000000000
885457070 18257718 927615960 3...

correct output
5000050000

user output
(empty)

Test 16

Group: 3

Verdict:

input
100000 50000
1 2 3 4 5 6 7 8 9 10 11 12 13 ...

correct output
3750075000

user output
(empty)

Test 17

Group: 3

Verdict:

input
100000 50000
100000 99999 99998 99997 99996...

correct output
3750075000

user output
(empty)