4519 2023. 03. 29 12:11:52 Eyusie Tűzijáték cpp17 Elfogadva 50/50 17ms 6532 KiB
#include <bits/stdc++.h>

using namespace std;

int n, telh, tav;
vector<int> kapcs;
vector<int> varos;



int main()
{
    iostream::sync_with_stdio(0);
    cin.tie(0);
    cin >> n >> telh >> tav;
    kapcs.assign(n, 100001);
    int ertek;
    int counter = 0;
    for(int i = 0; i < n; i++)
    {
        cin >> ertek;
        varos.push_back(ertek);
    }
    set<int> varosok;
    int current = telh-1;
    for(int i = telh-2; i >= 0; i--)
    {
        if(varos[current] - varos[i] >= tav)
        {
            varosok.insert(i+1);
            counter++;
            current = i;
        }
    }
    current = telh-1;
    for(int i = telh; i < varos.size(); i++)
    {
        if(varos[i] - varos[current] >= tav)
        {
            varosok.insert(i+1);
            counter++;
            current = i;
        }
    }
        varosok.insert(telh);
        cout << varosok.size() << "\n";
        for(set<int>::iterator itr = varosok.begin(); itr != varosok.end(); ++itr)
        {
            cout << *itr << " ";
        }
}
Részfeladat Összpont Teszt Verdikt Idő Memória
base 50/50
1 Elfogadva 0/0 3ms 1832 KiB
2 Elfogadva 0/0 17ms 4344 KiB
3 Elfogadva 2/2 2ms 2136 KiB
4 Elfogadva 2/2 2ms 2140 KiB
5 Elfogadva 2/2 3ms 2264 KiB
6 Elfogadva 2/2 2ms 2368 KiB
7 Elfogadva 2/2 3ms 2488 KiB
8 Elfogadva 2/2 3ms 3004 KiB
9 Elfogadva 2/2 3ms 3112 KiB
10 Elfogadva 2/2 4ms 3520 KiB
11 Elfogadva 2/2 4ms 3460 KiB
12 Elfogadva 2/2 4ms 3936 KiB
13 Elfogadva 2/2 4ms 3880 KiB
14 Elfogadva 2/2 4ms 3940 KiB
15 Elfogadva 3/3 4ms 4000 KiB
16 Elfogadva 3/3 7ms 4388 KiB
17 Elfogadva 3/3 8ms 4760 KiB
18 Elfogadva 3/3 8ms 5244 KiB
19 Elfogadva 3/3 17ms 6532 KiB
20 Elfogadva 3/3 17ms 6404 KiB
21 Elfogadva 4/4 17ms 6496 KiB
22 Elfogadva 4/4 14ms 6116 KiB