Alitaulukot

input/code.cpp: In function 'int main()':
input/code.cpp:38:20: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  for (int a = 0; a < layers[0].size(); a++) {
                  ~~^~~~~~~~~~~~~~~~~~
input/code.cpp:42:21: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
   for (int b = 0; b < layers[a].size()-1; b++) {
                   ~~^~~~~~~~~~~~~~~~~~~~
input/code.cpp:53:12: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
   if (numf == flayer.size()) {
       ~~~~~^~~~~~~~~~~~~~~~

#include <algorithm>
#include <bits/stdc++.h>

#define MK(x, y) std::make_pair(x, y)

int main() {
	int n, k;
	std::cin>> n >> k;
	int res = n;
	std::vector<std::vector<std::pair<int,int> > > layers; // set first -1 for invalid

	if (n < 2) {
		std::cout<< n << std::endl;
		return 0;
	}

	int prev;
	std::cin>> prev;
	std::vector<std::pair<int,int> > flayer;
	for (int i = 0; i < n-1; i++) {
		int in;
		std::cin>> in;
		int max = std::max(prev, in), min = std::min(prev, in);
		if (max - min > k) {
			flayer.push_back(MK(-1, -1));
		} else {
			flayer.push_back(MK(max, min));
			res++;
		}
		prev = in;
	}
	layers.push_back(flayer);
	if (layers[0].size() < 2) {
		std::cout<< n << std::endl;
		return 0;
	}
	int numf = 0;
	for (int a = 0; a < layers[0].size(); a++) {
		flayer.clear();
		flayer.shrink_to_fit(); //instead of reinitializion
		numf = 0;
		for (int b = 0; b < layers[a].size()-1; b++) {
			int max = std::max(layers[a][b].first, layers[a][b+1].first);
			int min = std::min(layers[a][b].second, layers[a][b+1].second);
			if (max == -1 || max - min > k) {
				flayer.push_back(MK(-1, -1));
				numf++;
			} else {
				flayer.push_back(MK(max, min));
				res++;
			}
		}
		if (numf == flayer.size()) {
			break;
		}
		layers.push_back(flayer);
	}

	std::cout<< res << std::endl;
	return 0;
}

Test 1

Group: 1, 2, 3

Verdict: ACCEPTED

Test 2

Group: 1, 2, 3

Verdict: ACCEPTED

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

Test 12

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 13

Group: 3

Verdict: TIME LIMIT EXCEEDED

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