Alitaulukot

#include <bits/stdc++.h>

using namespace std;


int main(){


    int n, k;
    cin >> n;
    cin >> k;
    vector<int> x;
    multiset<int> a;
    int in;
    for(int i = 0; i < n; i++){
        cin >> in;
        x.push_back(in);
    }
    
    long count = 0;

    int i = 0;
    int j = 0;

    int m = x[0];
    int M = x[0];
    int first;
    while(i<n && j<n){
        a.insert(x[j]);
        m = min(m, x[j]);
        M = max(M, x[j]);
        if(M-m>k){
            a.erase(a.find(x[j]));
            count += a.size();
            first = x[i];
            a.erase(a.find(first));
            i += 1;
            if(a.size() != 0){
                if(first == M){
                    M = *a.begin();
                }
                if(first == m){
                    m = *a.end();
                }
            }else{
                m = x[j];
                M = x[j];
            }
            continue;
        }
        if(j==n-1){
            count += (long) (a.size()*(a.size()+1)/2);
            break;
        }
        j++;
    }
    cout << count;
}

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

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

Test 15

Group: 3

Verdict: ACCEPTED

Test 16

Group: 3

Verdict: WRONG ANSWER

Test 17

Group: 3

Verdict: WRONG ANSWER