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];
            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 << std::endl;

    /*
    for (int bar = 1; bar <= size; bar++) {
        for (int i = 0; i < size - bar + 1; i++) {
            bool valid = true;
            int max = -1;
            int min = -1;
            for (int j = i; j < i + bar; j++) {
                int num = numbers[j];
                if (num > max) {
                    max = num;
                }
                if (num < min || min == -1) {
                    min = num;
                }
                if ((max - min) > k) {
                    valid = false;
                    break;
                }
            }
            if (valid) {
                count++;
            }
        }
        // std::cout << "Count was at " << count << " after bar " << bar << 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