147512025-02-01 16:28:06mateTűzijátékcpp17Hibás válasz 48/5041ms928 KiB
#include <bits/stdc++.h>
using namespace std;

int main() {
	int n, k, d; cin >> n >> k >> d;
    vector <int> v(n+1);
    for(int i = 0; i < n; i++)  cin >> v[i];
    vector <int> ans;
    int l = 0;
    if(v[k-1] - v[0] >= d || k == 1){
        l = v[0];
        ans.push_back(1);
    }
    
    for(int i = 1; i < n; i++){
        if(v[i] - l >= d && v[i] + d <= v[k-1]){
            ans.push_back(i+1);
            l = v[i];
        }else{
            continue;
        }
    }
    if(k != 1){
        l = v[k-1];
        ans.push_back(k);
    }
    
    for(int i = k + 1; i < n; i++){
        if(v[i] - l >= d){
            ans.push_back(i+1);
            l = v[i];
        }
    }
    cout << ans.size() << '\n';
    for(int x : ans)    cout << x << ' ';
}
RészfeladatÖsszpontTesztVerdiktIdőMemória
base48/50
1Elfogadva0/01ms316 KiB
2Elfogadva0/041ms820 KiB
3Hibás válasz0/21ms316 KiB
4Elfogadva2/21ms316 KiB
5Elfogadva2/21ms500 KiB
6Elfogadva2/21ms316 KiB
7Elfogadva2/21ms316 KiB
8Elfogadva2/21ms316 KiB
9Elfogadva2/23ms316 KiB
10Elfogadva2/24ms316 KiB
11Elfogadva2/24ms316 KiB
12Elfogadva2/27ms316 KiB
13Elfogadva2/28ms500 KiB
14Elfogadva2/27ms472 KiB
15Elfogadva3/37ms464 KiB
16Elfogadva3/312ms516 KiB
17Elfogadva3/316ms572 KiB
18Elfogadva3/318ms600 KiB
19Elfogadva3/341ms820 KiB
20Elfogadva3/341ms820 KiB
21Elfogadva4/441ms928 KiB
22Elfogadva4/441ms800 KiB