13152022-05-11 19:19:48nkdorka1212Tűzijátékcpp11Elfogadva 50/5019ms8288 KiB
#include <bits/stdc++.h>

using namespace std;
#pragma GCC target ("avx2")
#pragma GCC optimization ("O3")
#pragma GCC optimization ("unroll-loops")

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    int n,s,k;
    cin>>n>>s>>k;
    vector<int>v(n+1);
    vector<bool>ok(n+1,0);
    for(int i=1;i<=n;i++)
    {
        cin>>v[i];
    }
    ok[s]=true;
    int elozo=v[s];
    for(int i=s+1;i<=n;i++)
    {
        if(k<=v[i]-elozo)
        {
            ok[i]=true;
            elozo=v[i];
        }
    }
    elozo=v[s];
    for(int i=s-1;i>=1;i--)
    {
        if(k<=elozo-v[i])
        {
            ok[i]=true;
            elozo=v[i];
        }
    }
    vector<int>mo;
    for(int i=1;i<=n;i++)
    {
        if(ok[i])
        {
            mo.push_back(i);
        }
    }
    cout<<mo.size()<<'\n';
    for(int x:mo)
    {
        cout<<x<<" ";
    }

    return 0;
}
RészfeladatÖsszpontTesztVerdiktIdőMemória
base50/50
1Elfogadva0/02ms1852 KiB
2Elfogadva0/014ms3752 KiB
3Elfogadva2/21ms2700 KiB
4Elfogadva2/21ms2728 KiB
5Elfogadva2/21ms2704 KiB
6Elfogadva2/21ms2708 KiB
7Elfogadva2/21ms2712 KiB
8Elfogadva2/21ms2728 KiB
9Elfogadva2/23ms2756 KiB
10Elfogadva2/23ms2808 KiB
11Elfogadva2/22ms3000 KiB
12Elfogadva2/23ms3100 KiB
13Elfogadva2/24ms3360 KiB
14Elfogadva2/23ms3360 KiB
15Elfogadva3/33ms3480 KiB
16Elfogadva3/36ms3920 KiB
17Elfogadva3/36ms4164 KiB
18Elfogadva3/37ms4580 KiB
19Elfogadva3/314ms6120 KiB
20Elfogadva3/313ms6848 KiB
21Elfogadva4/418ms7652 KiB
22Elfogadva4/419ms8288 KiB