1315 2022. 05. 11 19:19:48 nkdorka1212 Tűzijáték cpp11 Elfogadva 50/50 19ms 8288 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 Összpont Teszt Verdikt Idő Memória
base 50/50
1 Elfogadva 0/0 2ms 1852 KiB
2 Elfogadva 0/0 14ms 3752 KiB
3 Elfogadva 2/2 1ms 2700 KiB
4 Elfogadva 2/2 1ms 2728 KiB
5 Elfogadva 2/2 1ms 2704 KiB
6 Elfogadva 2/2 1ms 2708 KiB
7 Elfogadva 2/2 1ms 2712 KiB
8 Elfogadva 2/2 1ms 2728 KiB
9 Elfogadva 2/2 3ms 2756 KiB
10 Elfogadva 2/2 3ms 2808 KiB
11 Elfogadva 2/2 2ms 3000 KiB
12 Elfogadva 2/2 3ms 3100 KiB
13 Elfogadva 2/2 4ms 3360 KiB
14 Elfogadva 2/2 3ms 3360 KiB
15 Elfogadva 3/3 3ms 3480 KiB
16 Elfogadva 3/3 6ms 3920 KiB
17 Elfogadva 3/3 6ms 4164 KiB
18 Elfogadva 3/3 7ms 4580 KiB
19 Elfogadva 3/3 14ms 6120 KiB
20 Elfogadva 3/3 13ms 6848 KiB
21 Elfogadva 4/4 18ms 7652 KiB
22 Elfogadva 4/4 19ms 8288 KiB