155122025-02-20 09:54:29999Tűzijátékcpp17Accepted 50/5045ms820 KiB
// Source: https://usaco.guide/general/io

#include <bits/stdc++.h>
using namespace std;

int main() {
    int n,in,k;cin>>n>>in>>k;
    vector<int> v(n+1),ans;
    for(int i = 1; i<=n;i++)cin>>v[i];
    int lasti=in;
    ans.push_back(in);
    for(int i = in+1;i<=n;i++){
        if(v[i]-v[lasti]>=k){
            lasti=i;
            ans.push_back(i);
        }
    }
    lasti=in;
    for(int i =in-1;i>0;i--){
        if(v[lasti]-v[i]>=k){
            lasti=i;
            ans.push_back(i);
        }
    }
    sort(ans.begin(),ans.end());
    cout<<ans.size()<<endl;
    for(int i : ans)cout<<i<<' ';
}
SubtaskSumTestVerdictTimeMemory
base50/50
1Accepted0/01ms316 KiB
2Accepted0/043ms820 KiB
3Accepted2/21ms500 KiB
4Accepted2/21ms316 KiB
5Accepted2/21ms316 KiB
6Accepted2/21ms316 KiB
7Accepted2/21ms316 KiB
8Accepted2/21ms316 KiB
9Accepted2/23ms316 KiB
10Accepted2/24ms444 KiB
11Accepted2/24ms316 KiB
12Accepted2/28ms316 KiB
13Accepted2/28ms316 KiB
14Accepted2/27ms316 KiB
15Accepted3/37ms468 KiB
16Accepted3/312ms316 KiB
17Accepted3/316ms564 KiB
18Accepted3/319ms604 KiB
19Accepted3/345ms820 KiB
20Accepted3/343ms820 KiB
21Accepted4/443ms800 KiB
22Accepted4/443ms800 KiB