238382026-01-30 16:29:37Rokus23Tűzijátékcpp17Accepted 50/5016ms948 KiB
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
using pii = pair<int,int>;

int main() {
	ios::sync_with_stdio(false);
    cin.tie(0);
	int N,K,d;cin>>N>>K>>d;
	vector<int> v(N);
	for(int i=0;i<N;i++) cin>>v[i];
	K--;
	int last=v[K];
	vector<int> ans;
	ans.push_back(K+1);
	for(int i=K;i<N;i++) {
		if(v[i]-last>=d) {
			ans.push_back(i+1);
			last=v[i];
		}
	}
	last=v[K];
	for(int i=K;i>=0;i--) {
		if(last-v[i]>=d) {
			ans.push_back(i+1);
			last=v[i];
		}
	}
	cout<<ans.size()<<"\n";
	sort(ans.begin(),ans.end());
	for(int i=0;i<ans.size();i++) cout<<ans[i]<<" ";
}
SubtaskSumTestVerdictTimeMemory
base50/50
1Accepted0/01ms512 KiB
2Accepted0/016ms832 KiB
3Accepted2/21ms508 KiB
4Accepted2/21ms316 KiB
5Accepted2/21ms316 KiB
6Accepted2/21ms316 KiB
7Accepted2/21ms316 KiB
8Accepted2/21ms352 KiB
9Accepted2/22ms500 KiB
10Accepted2/22ms316 KiB
11Accepted2/22ms500 KiB
12Accepted2/23ms316 KiB
13Accepted2/23ms508 KiB
14Accepted2/23ms316 KiB
15Accepted3/33ms316 KiB
16Accepted3/34ms564 KiB
17Accepted3/36ms624 KiB
18Accepted3/38ms836 KiB
19Accepted3/316ms836 KiB
20Accepted3/316ms948 KiB
21Accepted4/414ms824 KiB
22Accepted4/414ms840 KiB