Alitaulukot

//  Datatähti 2021 Alku
//  C Alitaulukot

#include <iostream>

using namespace std;

int main(int argc, char **argv) {
    int n, k;
    cin >> n >> k;
    
    auto a = new int[n];
    
    for (int x,i=0; i<n; i++) {
        if (!scanf("%d", &x))
            break;
        a[i] = x;
    }
    
    unsigned long long c;
    c = n*(n+1)/2;
    

    
    for (int m=-1, p=1, i=0; i<n-1; i++) {
        if (p == i)
            p++;
            
        for (int j=p; j<n; j++ ){
            if (a[i] > a[j]+k || a[j] > a[i]+k){
                c -= (n-j)*(i-m);
                m = i;
                p = j;
                break;
            }
        }
    }
    
    cout << c << "\n";
}

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

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

Test 10

Group: 2, 3

Verdict: ACCEPTED

Test 11

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 12

Group: 3

Verdict: WRONG ANSWER

Test 13

Group: 3

Verdict: WRONG ANSWER

Test 14

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 15

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 16

Group: 3

Verdict: WRONG ANSWER

Test 17

Group: 3

Verdict: WRONG ANSWER