CSES - Datatähti 2021 alku - Results
Submission details
Task:Alitaulukot
Sender:Anniiiz
Submission time:2020-09-30 16:59:31 +0300
Language:CPython3
Status:READY
Result:0
Feedback
groupverdictscore
#10
#20
#30
Test results
testverdicttimegroup
#1ACCEPTED0.03 s1, 2, 3details
#2ACCEPTED0.02 s1, 2, 3details
#30.02 s1, 2, 3details
#40.02 s1, 2, 3details
#50.02 s1, 2, 3details
#6ACCEPTED0.64 s2, 3details
#7ACCEPTED0.03 s2, 3details
#80.03 s2, 3details
#90.03 s2, 3details
#100.03 s2, 3details
#11--3details
#12ACCEPTED0.35 s3details
#130.23 s3details
#14--3details
#15--3details
#16--3details
#17--3details

Code

import math


def main():
  d = input()
  luvut = d.split(" ")
  n = int(luvut[0])
  k = int(luvut[1])
  del luvut
  b = input()
  taulukko = b.split(" ")
  s = 0
  a = 0
  maara = 0
  pienin = int(taulukko[0])
  suurin = int(taulukko[0])
  joku = list(taulukko[0])
  while s < n-1:
    if a != n - 1:
      if abs(suurin - int(taulukko[a + 1])) <= k and abs(pienin - int(taulukko[a + 1])) <= k:
        a += 1
        joku.append(taulukko[a])
        suurin = max(suurin, int(taulukko[a]))
        pienin = min(pienin, int(taulukko[a]))
      else:
        maara += len(joku)
        if len(joku) > 1:
          if int(joku[0]) == pienin:
            joku.pop(0)
            pienin = valitsePienin(joku)
          elif int(joku[0]) == suurin:
            joku.pop(0)
            suurin = valitseSuurin(joku)
          else:
            joku.pop(0)
        else:
          joku.pop(0)
        s += 1
        if s > a:
          joku.append(taulukko[s])
          pienin = int(taulukko[s])
          suurin = int(taulukko[s])
          a = s
    else:
      maara += len(joku)
      if len(joku) > 1:
        if int(joku[0]) == pienin:
          joku.pop(0)
          pienin = valitsePienin(joku)
        elif int(joku[0]) == suurin:
          joku.pop(0)
          suurin = valitseSuurin(joku)
        else:
          joku.pop(0)
      else:
        joku.pop(0)
      s += 1
  maara += 1
  print(maara)


def valitsePienin(lista):
  pienin = int(lista[0])
  for i in lista:
    pienin = min(pienin, int(i))
  return pienin

def valitseSuurin(lista):
  suurin = int(lista[0])
  for i in lista:
    suurin = max(suurin, int(i))
  return suurin



main()

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: ACCEPTED

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

correct output
317

user output
317

Test 3

Group: 1, 2, 3

Verdict:

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

correct output
119

user output
218

Test 4

Group: 1, 2, 3

Verdict:

input
100 990000000
111122929 961821360 578238211 ...

correct output
4006

user output
4566

Test 5

Group: 1, 2, 3

Verdict:

input
100 1000000000
553190572 453407680 667300705 ...

correct output
5050

user output
5842

Test 6

Group: 2, 3

Verdict: ACCEPTED

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

correct output
2001000

user output
2001000

Test 7

Group: 2, 3

Verdict: ACCEPTED

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

correct output
6340

user output
6340

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
4392

Test 9

Group: 2, 3

Verdict:

input
2000 999000000
746120950 772769620 721488968 ...

correct output
1287776

user output
1292528

Test 10

Group: 2, 3

Verdict:

input
2000 1000000000
621947980 510355354 756705418 ...

correct output
2001000

user output
2016992

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: ACCEPTED

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

correct output
317066

user output
317066

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
222187

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)