Alitaulukot

#include <bits/stdc++.h>

using namespace std;


int main(){

    vector<long> x;
    long n, k;

    cin >> n;
    cin >> k;
    long in;
    for(long i = 0; i < n; i++){
        cin >> in;
        x.push_back(in);
    }
    
    long i = 0;
    long j = 0;
    
    vector<long> a;
    long count = 0;
    long m = x[0];
    long M = x[0];
    while(i < n && j < n){
        a.push_back(x[j]);
        m = min(m, x[j]);
        M = max(M, x[j]);
        if(M-m > k){
            a.pop_back();
           
            count += a.size();
            
            a.erase(a.begin());
            
            if(a.size() != 0){
                
                m = *min_element(a.begin(), a.end()); //o(log n)
                M = *max_element(a.begin(), a.end());
            }
            else{
                
                m = x[j];
                M = x[j];
            }
            i++;
            
            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: 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: ACCEPTED

Test 7

Group: 2, 3

Verdict: ACCEPTED

Test 8

Group: 2, 3

Verdict: ACCEPTED

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

Test 13

Group: 3

Verdict: ACCEPTED

Test 14

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 15

Group: 3

Verdict: ACCEPTED

Test 16

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 17

Group: 3

Verdict: TIME LIMIT EXCEEDED