42732023-03-21 10:14:32CWMTűzijátékcpp17Accepted 50/5045ms4860 KiB
// Gyak_Tuzijatek.cpp : This file contains the 'main' function. Program execution begins and ends there.
//

#include <iostream>
#include <vector>
using namespace std;

int main()
{
    int n, pos, k;
    cin >> n >> pos >> k;
	vector<int> dist;
	for (size_t i = 0; i < n; i++)
	{
		int a;
		cin >> a;
		dist.push_back(a);
	}
	int curSmallest = dist[pos - 1];
	vector<int> more;
	vector<int> less;
	for (int i = pos; i < n; i++)
	{
		if (dist[i] >= curSmallest+k) {
			curSmallest = dist[i];
			more.push_back(i);
		}
	}
	curSmallest = dist[pos - 1];
	for (int i = pos-2; i >= 0; i--)
	{
		if (dist[i]+k <= curSmallest) {
			curSmallest = dist[i];
			less.push_back(i);
		}
	}
	cout << less.size() + more.size() + 1 << "\n";
	for (size_t i = 0; i < less.size(); i++)
	{
		cout << less[less.size()-i-1]+1 << " ";
	}
	cout << pos << " ";
	for (size_t i = 0; i < more.size(); i++)
	{
		cout << more[i]+1 << " ";
	}
}

// Run program: Ctrl + F5 or Debug > Start Without Debugging menu
// Debug program: F5 or Debug > Start Debugging menu

// Tips for Getting Started: 
//   1. Use the Solution Explorer window to add/manage files
//   2. Use the Team Explorer window to connect to source control
//   3. Use the Output window to see build output and other messages
//   4. Use the Error List window to view errors
//   5. Go to Project > Add New Item to create new code files, or Project > Add Existing Item to add existing code files to the project
//   6. In the future, to open this project again, go to File > Open > Project and select the .sln file
SubtaskSumTestVerdictTimeMemory
base50/50
1Accepted0/03ms1812 KiB
2Accepted0/043ms3132 KiB
3Accepted2/23ms2220 KiB
4Accepted2/23ms2304 KiB
5Accepted2/23ms2428 KiB
6Accepted2/22ms2504 KiB
7Accepted2/23ms2620 KiB
8Accepted2/23ms2512 KiB
9Accepted2/24ms2820 KiB
10Accepted2/26ms3080 KiB
11Accepted2/26ms3036 KiB
12Accepted2/28ms3228 KiB
13Accepted2/29ms3532 KiB
14Accepted2/28ms3580 KiB
15Accepted3/38ms3724 KiB
16Accepted3/314ms3952 KiB
17Accepted3/317ms4220 KiB
18Accepted3/321ms4220 KiB
19Accepted3/345ms4732 KiB
20Accepted3/343ms4740 KiB
21Accepted4/443ms4736 KiB
22Accepted4/443ms4860 KiB