242122026-02-06 14:43:14ubormaciTűzijátékcpp17Accepted 50/5017ms1996 KiB
#include <iostream>
#include <algorithm> // for sort, mainly
#include <vector>
#include <map>
#include <set>
#include <cmath>
#include <array>
#include <string>
#include <cstdio>
#include <iterator>
#include <unordered_set>
#include <cstdint> // for int64_t, int32_t, etc
#include <queue>
#include <stack>
#include <deque>
#include <numeric> // gcd, lcm
#include <fstream>
#include <bitset> // for bitset
#include <iomanip>
#include <cassert> // for set with custom ordering
#include <type_traits> // for set with custom ordering
#include <utility> // for swap, forward, etc
using namespace std;

#pragma GCC optimize("O2")
// #pragma GCC optimize("O1","O2","O3","Ofast","unroll-loops")
// #pragma GCC target("sse","sse2","sse3","sse4.1","sse4.2","avx","avx2","fma")

template<typename A, typename B> ostream& operator<<(ostream &os, const pair<A, B> &p) { return os << '(' << p.first << ", " << p.second << ')'; }
template<typename T_container, typename T = typename enable_if<!is_same<T_container, string>::value, typename T_container::value_type>::type> ostream& operator<<(ostream &os, const T_container &v) { os << '{'; string sep; for (const T &x : v) os << sep << x, sep = ", "; return os << '}'; }
void dbg_out() { cout << endl; }
template<typename Head, typename... Tail> void dbg_out(Head H, Tail... T) { cout << ' ' << H; dbg_out(T...); }
#ifdef LOCAL
#define dbg(...) cout << "(" << #__VA_ARGS__ << "):", dbg_out(__VA_ARGS__)
#else
#define dbg(...)
#endif

/*

notes:

int64_t
stoi(string s) -> string to int
to_string() -> int (or else) to string

vector declaration:
vector<ll> v(n, 0)
vector<vector<ll>> v(n, vector<ll>(n, 0));

{if statement} ? {truth value} : {false value}

#ifdef LOCAL
    freopen("in.txt","r",stdin);
    freopen("out.txt","w",stdout);
#endif

std::lcm(ll a, ll b), std::gcd(int a, int b)

cout << fixed << setprecision(n);

set with custom ordering
set<ll, decltype(&cmp)> qu(cmp);

*/

typedef int64_t ll;

void solve() {

	ll n, s, d;
	cin >> n >> s >> d;
	vector<ll> v(n, 0);
	for(ll i = 0; i < n; i++) {
		cin >> v[i];
	}

	s--;
	vector<ll> last(n, -1);
	last[s] = v[s];

	for(ll j = s-1; j >= 0; j--) {
		
		if(last[j+1] - v[j] >= d) {
			last[j] = v[j];
		}else{
			last[j] = last[j+1];
		}
	}

	for(ll j = s+1; j < n; j++) {

		if(v[j] - last[j-1] >= d) {
			last[j] = v[j];
		}else{
			last[j] = last[j-1];
		}
	}

	//cerr << "\nlast=" << last;
  
	ll ans = 0;
	for(ll i = 0; i < n; i++) {
		if(last[i] == v[i]) {
			ans++;
		}
	}
	cout << ans << "\n";
	for(ll i = 0; i < n; i++) {
		if(last[i] == v[i]) {
			cout << i+1 << " ";
		}
	}

}

int main()
{
	std::ios_base::sync_with_stdio(false);
	//cin.tie(nullptr);
	//cout.tie(nullptr);

	solve();

	return 0;
}
SubtaskSumTestVerdictTimeMemory
base50/50
1Accepted0/02ms316 KiB
2Accepted0/017ms1996 KiB
3Accepted2/21ms316 KiB
4Accepted2/22ms316 KiB
5Accepted2/21ms316 KiB
6Accepted2/21ms540 KiB
7Accepted2/21ms316 KiB
8Accepted2/21ms316 KiB
9Accepted2/22ms316 KiB
10Accepted2/22ms552 KiB
11Accepted2/22ms552 KiB
12Accepted2/24ms748 KiB
13Accepted2/24ms564 KiB
14Accepted2/23ms564 KiB
15Accepted3/34ms580 KiB
16Accepted3/34ms820 KiB
17Accepted3/37ms1008 KiB
18Accepted3/38ms1084 KiB
19Accepted3/317ms1992 KiB
20Accepted3/317ms1988 KiB
21Accepted4/416ms1916 KiB
22Accepted4/416ms1844 KiB