Alitaulukot

def muodostaYlitaulukot(gayArray, n, k):
    ylitaulukkoLista = []
    tPointer = 0
    ePointer = 1
    lastEPointer = -1
    while ePointer < n: #niin kauan kuin taulukkoa ei olla käyty loppuun
        ylitaulukko = {}
        ylitaulukko["alku"] = tPointer
        ylitaulukko["loppu"] = tPointer
        ylitaulukko["maxVal"] = gayArray[tPointer]
        ylitaulukko["minVal"] = gayArray[tPointer]

        while (gayArray[ePointer] <= ylitaulukko["minVal"] + k) and (gayArray[ePointer] >= ylitaulukko["maxVal"] - k): #niin kauan kun tuleva lisäys on laillinen

            ylitaulukko["maxVal"] = max(ylitaulukko["maxVal"], gayArray[ePointer])  #update records
            ylitaulukko["minVal"] = min(ylitaulukko["minVal"], gayArray[ePointer])
            ePointer += 1
            if ePointer == n:
                break

        # LISÄYS EI OLE ENÄÄN LAILLINEN
        # päätepiste eri meinaa uutta taulukkoa:
        if ePointer != lastEPointer:
            ylitaulukko["loppu"] = ePointer - 1
            ylitaulukkoLista.append(ylitaulukko)
            lastEPointer = ePointer

        tPointer += 1
        ePointer = max(ePointer, tPointer + 1)

    if ylitaulukkoLista[len(ylitaulukkoLista) - 1]["loppu"] != n - 1:
        ylitaulukko = {}
        ylitaulukko["alku"] = tPointer
        ylitaulukko["loppu"] = tPointer
        ylitaulukkoLista.append(ylitaulukko)

    return ylitaulukkoLista



def laskeAlitaulukot(ylitaulukkoLista, n):  #[{alku: x, loppu: y}, {}]
    result = 0
    lastEnd = -1
    for ylitaulukko in ylitaulukkoLista:
        pituus = ylitaulukko["loppu"] - ylitaulukko["alku"] + 1
        result += (pituus*(pituus+1))//2
        if lastEnd >= ylitaulukko["alku"]: #jos on toisen taulokon päällä niin vähennä overlap
            pituus = lastEnd - ylitaulukko["alku"] + 1
            result -= (pituus*(pituus+1))//2
        lastEnd = ylitaulukko["loppu"]

    return result

def main():
    n, k = [int(i) for i in input().split(" ")]
    gayArray = [int(i) for i in input().split(" ")]
    if n == 1:
        print("1")
    else:
        print(laskeAlitaulukot(muodostaYlitaulukot(gayArray, n, k), n))

main()

Test 1

Group: 1, 2, 3

Verdict: ACCEPTED

Test 2

Group: 1, 2, 3

Verdict: WRONG ANSWER

Test 3

Group: 1, 2, 3

Verdict: ACCEPTED

Test 4

Group: 1, 2, 3

Verdict: WRONG ANSWER

Test 5

Group: 1, 2, 3

Verdict: ACCEPTED

Test 6

Group: 2, 3

Verdict: ACCEPTED

Test 7

Group: 2, 3

Verdict: WRONG ANSWER

Test 8

Group: 2, 3

Verdict: WRONG ANSWER

Test 9

Group: 2, 3

Verdict: WRONG ANSWER

Test 10

Group: 2, 3

Verdict: ACCEPTED

Test 11

Group: 3

Verdict: ACCEPTED

Test 12

Group: 3

Verdict: WRONG ANSWER

Test 13

Group: 3

Verdict: WRONG ANSWER

Test 14

Group: 3

Verdict: WRONG ANSWER

Test 15

Group: 3

Verdict: ACCEPTED

Test 16

Group: 3

Verdict: ACCEPTED

Test 17

Group: 3

Verdict: ACCEPTED