4298 | 2023. 03. 22 14:37:40 | horvathabel | Tűzijáték | cpp17 | Accepted 50/50 | 46ms | 6664 KiB |
#include <bits/stdc++.h>
using namespace std;
using ll=long long;
int main() {
int n,k,s;
cin>>n>>k>>s;
vector<ll> tav={-50000001};
for (int i=0; i<n;i++){
ll c;
cin>>c;
tav.push_back(c);
}
vector<int> ans={k};
int mst=tav[k];
for (int i=k-1;i>0;i--){
if (tav[i]+s<=mst){
mst=tav[i];
ans.push_back(i);
}
}
mst=tav[k];
for (int i=k+1; i<=n;i++){
if (tav[i]-s>=mst){
mst=tav[i];
ans.push_back(i);
}
}
sort(ans.begin(), ans.end());
cout<<ans.size()<<endl;
for (int x:ans) cout<<x<<" ";
}
Subtask | Sum | Test | Verdict | Time | Memory | ||
---|---|---|---|---|---|---|---|
base | 50/50 | ||||||
1 | Accepted | 0/0 | 3ms | 1812 KiB | |||
2 | Accepted | 0/0 | 43ms | 4028 KiB | |||
3 | Accepted | 2/2 | 3ms | 2336 KiB | |||
4 | Accepted | 2/2 | 3ms | 2548 KiB | |||
5 | Accepted | 2/2 | 3ms | 2648 KiB | |||
6 | Accepted | 2/2 | 2ms | 2852 KiB | |||
7 | Accepted | 2/2 | 3ms | 3064 KiB | |||
8 | Accepted | 2/2 | 3ms | 3184 KiB | |||
9 | Accepted | 2/2 | 4ms | 3540 KiB | |||
10 | Accepted | 2/2 | 6ms | 3940 KiB | |||
11 | Accepted | 2/2 | 6ms | 4120 KiB | |||
12 | Accepted | 2/2 | 8ms | 4600 KiB | |||
13 | Accepted | 2/2 | 9ms | 4808 KiB | |||
14 | Accepted | 2/2 | 8ms | 4820 KiB | |||
15 | Accepted | 3/3 | 8ms | 4816 KiB | |||
16 | Accepted | 3/3 | 14ms | 4916 KiB | |||
17 | Accepted | 3/3 | 17ms | 5556 KiB | |||
18 | Accepted | 3/3 | 20ms | 5548 KiB | |||
19 | Accepted | 3/3 | 45ms | 6524 KiB | |||
20 | Accepted | 3/3 | 43ms | 6664 KiB | |||
21 | Accepted | 4/4 | 46ms | 6600 KiB | |||
22 | Accepted | 4/4 | 43ms | 6592 KiB |