Alitaulukot

#include <iostream>
#include <vector>
#include <algorithm>

int main() {
    int size;
    std::cin >> size;
    int k;
    std::cin >> k;
    std::vector<int> numbers = std::vector<int>();

    for (int i = 0; i < size; i++) {
        int n;
        std::cin >> n;
        numbers.push_back(n);
    }

    int count = 0;
    for (int i = 0; i < size; i++) {
        int min = -1;
        int max = -1;
        for (int j = i; j < (int) numbers.size(); j++) {
            int num = numbers[j];
            // std::cout << num << std::endl;
            if (num > max) {
                max = num;
            }
            if (num < min || min == -1) {
                min = num;
            }
            if ((max - min) > k) {
                count += (j - i) * (j - i + 1) / 2;
                i = j - 1;
                break;
            } else if (j + 1 == size) {
                count += (j - i + 1) * (j - i + 2) / 2;
                i = size;
                break;
            }
        }
        // std::cout << "Count is " << count << std::endl;
    }
    std::cout << count << std::endl;

    return 0;
}

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: WRONG ANSWER

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: WRONG ANSWER

Test 16

Group: 3

Verdict: WRONG ANSWER

Test 17

Group: 3

Verdict: WRONG ANSWER