Alitaulukot

import math


def main():
  a = input()
  luvut = a.split(" ")
  b = input()
  taulukko = b.split(" ")
  s = 0
  maara = 0
  while s < len(taulukko)-1:
    a = s
    pienin = int(taulukko[s])
    suurin = int(taulukko[s])
    while True:
      if a != len(taulukko) - 1:
        if abs(suurin - int(taulukko[a + 1])) <= int(luvut[1]) and abs(pienin - int(taulukko[a + 1])) <= int(luvut[1]):
          a += 1
          if int(taulukko[a]) > suurin:
            suurin = int(taulukko[a])
          if int(taulukko[a]) < pienin:
            pienin = int(taulukko[a])
        else:
          maara += a-s+1
          s += 1
          if int(taulukko[s]) > suurin:
            suurin = int(taulukko[s])
          if int(taulukko[s]) < pienin:
            pienin = int(taulukko[s])
          break
      else:
        maara += a-s+1
        s += 1
        if int(taulukko[s]) > suurin:
          suurin = int(taulukko[s])
        if int(taulukko[s]) < pienin:
          pienin = int(taulukko[s])
        break
  maara += 1
  print(maara)


main()

Test 1

Group: 1, 2, 3

Verdict: ACCEPTED

Test 2

Group: 1, 2, 3

Verdict: ACCEPTED

Test 3

Group: 1, 2, 3

Verdict: ACCEPTED

Test 4

Group: 1, 2, 3

Verdict: ACCEPTED

Test 5

Group: 1, 2, 3

Verdict: ACCEPTED

Test 6

Group: 2, 3

Verdict: TIME LIMIT EXCEEDED

Test 7

Group: 2, 3

Verdict: ACCEPTED

Test 8

Group: 2, 3

Verdict: ACCEPTED

Test 9

Group: 2, 3

Verdict: TIME LIMIT EXCEEDED

Test 10

Group: 2, 3

Verdict: TIME LIMIT EXCEEDED

Test 11

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 12

Group: 3

Verdict: ACCEPTED

Test 13

Group: 3

Verdict: ACCEPTED

Test 14

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 15

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 16

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 17

Group: 3

Verdict: TIME LIMIT EXCEEDED