Alitaulukot

class Ylitaulukko:
    def __init__(self, start, end):
        self.start = start
        self.end = end

def muodostaYlitaulukot(gayArray, n, k):
    tPointer = 0
    ePointer = 1
    lastEPointer = -1
    maxVal = gayArray[0]
    minVal = gayArray[0]
    maxLoc = 0
    minLoc = 0
    ylitaulukot = []

    while ePointer < n:
        while (ePointer < n) and (gayArray[ePointer] <= minVal + k) and (gayArray[ePointer] >= maxVal - k):
            if gayArray[ePointer] >= maxVal:
                maxVal = gayArray[ePointer]
                maxLoc = ePointer
            elif gayArray[ePointer] <= minVal:
                minVal = gayArray[ePointer]
                minLoc = ePointer
            ePointer += 1

        # ePointer ei olisi enää laillinen lisäys
        if ePointer != lastEPointer:
            ylitaulukot.append(Ylitaulukko(tPointer, ePointer - 1))
            lastEPointer = ePointer

        if ePointer == n:
            break

        if (gayArray[ePointer] <= minVal + k):
            tPointer = maxLoc+1
        else:
            tPointer = minLoc+1
            
        ePointer = max(ePointer, tPointer + 1)
        maxVal = max(gayArray[tPointer:ePointer+1])
        minVal = min(gayArray[tPointer:ePointer+1])

    if ylitaulukot[-1].end != n-1:
        ylitaulukot.append(Ylitaulukko(n-1, n-1))

    return ylitaulukot



def laskeAlitaulukot(ylitaulukot):
    result = 0
    lastEnd = -1
    for ylitaulukko in ylitaulukot:
        length = ylitaulukko.end - ylitaulukko.start + 1
        result += (length*(length+1))/2

        if lastEnd >= ylitaulukko.start:
            length = lastEnd - ylitaulukko.start + 1
            result -= (length*(length+1))/2

        lastEnd = ylitaulukko.end

    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(int(laskeAlitaulukot(muodostaYlitaulukot(gayArray, n, k))))

main()

Test 1

Group: 1, 2, 3

Verdict: ACCEPTED

Test 2

Group: 1, 2, 3

Verdict: TIME LIMIT EXCEEDED

Test 3

Group: 1, 2, 3

Verdict: TIME LIMIT EXCEEDED

Test 4

Group: 1, 2, 3

Verdict: ACCEPTED

Test 5

Group: 1, 2, 3

Verdict: ACCEPTED

Test 6

Group: 2, 3

Verdict: ACCEPTED

Test 7

Group: 2, 3

Verdict: TIME LIMIT EXCEEDED

Test 8

Group: 2, 3

Verdict: TIME LIMIT EXCEEDED

Test 9

Group: 2, 3

Verdict: ACCEPTED

Test 10

Group: 2, 3

Verdict: ACCEPTED

Test 11

Group: 3

Verdict: ACCEPTED

Test 12

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 13

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 14

Group: 3

Verdict: ACCEPTED

Test 15

Group: 3

Verdict: ACCEPTED

Test 16

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 17

Group: 3

Verdict: TIME LIMIT EXCEEDED