#include<iostream>
#include<conio.h>
#include<windows.h>
#include<set>

using namespace std;
set<string>lista_uczniow;
set<string>::iterator it;
void menu()
{
    cout<<"Lista uczniow\n[1] Dodanie ucznia do listy\n[2] Wyswietlenie listy uczniow\n[3] Usuniecie ucznia z listy\n[0] Wyjscie z programu\n"<<"Twoj wybor: ";
}
void dodawanie(string y)
{
    lista_uczniow.insert(y);
}
void wyswietlanie()
{
    int i=1;
    if(lista_uczniow.empty()){
        cout<<"Nie ma uczniow na liscie. Nacisnij dowolny klawisz aby kontynuowac";
        getch();}
    else{
    cout<<"Lista to: \n";
    for(it=lista_uczniow.begin(); it!=lista_uczniow.end(); ++it)
    {
        cout<<i++<<". "<<*it<<"\n";
    }
    cout<<"Nacisnij dowolny klawisz aby kontynuowac";
    getch();
    }


}
void usuwanie(string x)
{

    if(lista_uczniow.empty()){
        cout<<"Nie ma uczniow na liscie. Nacisnij dowolny klawisz aby kontynuowac";
        getch();}
    else{
        lista_uczniow.erase(x);
        cout<<"Pomyslnie usunieto, powracanie do menu.";
        Sleep(500);
        system("cls");
    }

}



int main()
{
    int a;
    string b;
    string c;
    menu();
    a=getch();
    switch (a){
    case '1':
        system("cls");
        cout<<"Wybrales dodawanie";
        Sleep(500);
        system("cls");
        cout<<"Wprowadz ucznia ktorego chcesz dodac do listy: ";
        cin>>b;
        system("cls");
        dodawanie(b);
        system("cls");
        cout<<"Pomyslnie dodano, powracanie do menu.";
        Sleep(500);
        system("cls");
        main();
        break;
    case '2':
        system("cls");
        cout<<"Wybrales wyswietlanie listy uczniow";
        Sleep(500);
        system("cls");
        wyswietlanie();
        system("cls");
        main();
        break;
    case '3':
        system("cls");
        cout<<"Wybrales usuniecie elementu";
        Sleep(500);
        system("cls");
        cout<<"Wprowadz ucznia ktorego chcesz usunac: ";
        cin>>c;
        system("cls");
        usuwanie(c);
        main();
        break;
    case '0':
        exit(0);
        break;
    default:
        cout<<"Nie ma takiego wyboru";
        Sleep(500);
        system("cls");
        main();
        break;
    }

}