2047 2022. 12. 15 21:42:47 TomaSajt Tűzijáték cpp17 Accepted 50/50 14ms 4248 KiB
#include <bits/stdc++.h>
#define speed ios::sync_with_stdio(0);cin.tie(0)
using namespace std;

int main() {
    speed;
    int n, s, t;
    cin >> n >> s >> t;
    vector<int> v(n + 1);
    for (int i = 1; i <= n; i++) cin >> v[i];
    vector<int> res;
    res.push_back(s);
    int lastPos = v[s];
    for (int i = s - 1; i >= 1; i--) {
        if (lastPos - v[i] < t) continue;
        lastPos = v[i];
        res.push_back(i);
    }
    lastPos = v[s];
    for (int i = s + 1; i <= n; i++) {
        if (v[i] - lastPos < t) continue;
        lastPos = v[i];
        res.push_back(i);
    }
    sort(res.begin(), res.end());
    cout << res.size() << '\n';
    for (auto a : res) cout << a << ' ';
}
Subtask Sum Test Verdict Time Memory
base 50/50
1 Accepted 0/0 3ms 1700 KiB
2 Accepted 0/0 14ms 2960 KiB
3 Accepted 2/2 2ms 2068 KiB
4 Accepted 2/2 2ms 2276 KiB
5 Accepted 2/2 2ms 2368 KiB
6 Accepted 2/2 2ms 2372 KiB
7 Accepted 2/2 2ms 2492 KiB
8 Accepted 2/2 2ms 2552 KiB
9 Accepted 2/2 3ms 2568 KiB
10 Accepted 2/2 3ms 2904 KiB
11 Accepted 2/2 3ms 3044 KiB
12 Accepted 2/2 4ms 3164 KiB
13 Accepted 2/2 4ms 3424 KiB
14 Accepted 2/2 4ms 3180 KiB
15 Accepted 3/3 4ms 3184 KiB
16 Accepted 3/3 6ms 3308 KiB
17 Accepted 3/3 7ms 3192 KiB
18 Accepted 3/3 8ms 3492 KiB
19 Accepted 3/3 14ms 4016 KiB
20 Accepted 3/3 14ms 4248 KiB
21 Accepted 4/4 14ms 4216 KiB
22 Accepted 4/4 14ms 3912 KiB