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

    return 0;
}

Test 1

Group: 1, 2, 3

Verdict: WRONG ANSWER

Test 2

Group: 1, 2, 3

Verdict: WRONG ANSWER

Test 3

Group: 1, 2, 3

Verdict: WRONG ANSWER

Test 4

Group: 1, 2, 3

Verdict: WRONG ANSWER

Test 5

Group: 1, 2, 3

Verdict: WRONG ANSWER

Test 6

Group: 2, 3

Verdict: WRONG ANSWER

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

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: TIME LIMIT EXCEEDED

Test 17

Group: 3

Verdict: TIME LIMIT EXCEEDED