Vizualizare sursă pentru „3386 - Serbare 3”

== Enunt ==
Am o prietenă educatoare. Anul acesta are grupa mică și trebuie să organizeze serbarea de Crăciun. Ea s-a gândit să așeze copiii sub forma unui brad (cam așa ∆). Va împărți copiii în trei categorii: prima categorie va fi formată din primii X cei mai scunzi copii, a doua categorie este formată din copiii cu înălțimea maximă și restul Y de copii în a treia categorie. Copiii din prima categorie vor fi așezați în ordinea crescătoare a înălțimilor, iar cei din a treia categorie în ordinea descrescătoare a înălțimilor. Prietena mea vrea ca toți copiii cu aceeași înălțime să stea unul lângă altul, iar dacă numărul de copii din prima categorie nu poate fi egal cu cel din a treia, vrea ca prima categorie să fie mai numeroasă (X ≥ Y). Prietena mea știe N, numărul de elevi din grupă, prenumele copiilor și înălțimea lor în centimetri. Ea încearcă de câteva zile să îi așeze. Oare cum ar rezolva un elev olimpic la informatică această problemă?
== Cerinţa ==
Scrieţi un program care să determine:
1. Numărul maxim de copii cu aceeași înălțime.
2. O posibilă distribuire a înălțimilor sub forma unui brad.
3. Lista cu prenumele copiilor conform distribuirii înălțimilor determinată anterior.
== Date de intrare ==
Fișierul de intrare serbare.in conține pe prima linie cerința (1, 2 sau 3). Linia următoare conține un număr natural N reprezentând numărul de copii. Pe următoarele N linii conține câte o pereche de forma prenume înălțime, reprezentând prenumele și înălțimea corespunzătoare fiecărui copil din grupă.
== Date de ieșire ==
Fișierul de ieșire serbare.out va conține, dacă cerința este 1, o singură linie pe care va fi scris un număr natural determinat conform cerinței, dacă cerința este 2 va conține o linie pe care vor fi scrise N numere naturale determinate conform cerinței, dacă cerința este 3 va conține o linie pe care vor fi scrise prenumele copiilor conform distribuirii înălțimilor afișate la cerința 2.
== Restricţii şi precizări ==
* 0 < N, X, Y < 101; X ≥ Y
* 0 < înălțime < 151, numere naturale.
* prenumele este un cuvânt cu cel mult 50 de caractere alfabetice.
* NU există doi copii cu același prenume.
* Dacă sunt mai multe soluții se va afișa prima din punct de vedere lexicografic.
== Exemplul 1 ==
; serbare.in
 1
 5
 Ionel 96
 Vasile 88
 Gigel 66
 Mara 77
 Ana 77
; serbare.out
 2
<br>

== Explicație ==
Sunt 2 copii cu aceeași înălțime (77), acesta fiind numărul maxim de apariții a aceleiași înălțimi.
== Exemplul 2 ==
; serbare.in
 2
 5
 Ionel 96
 Vasile 88
 Gigel 66
 Mara 77
 Ana 77
; serbare.out
 66 77 77 96 88
<br>

== Explicație ==
Așezăm copiii în ordinea crescătoare a înălțimilor și apoi în ordinea descrescătoare a înălțimilor.
== Rezolvare ==
<syntaxhighlight lang="python" line>
def read_input(filename):
    with open(filename, 'r') as f:
        request = int(f.readline().strip())
        N = int(f.readline().strip())
        children = []
        for _ in range(N):
            name, height = f.readline().strip().split()
            children.append((name, int(height)))
    return request, children

def write_output(filename, output):
    with open(filename, 'w') as f:
        if isinstance(output, int):
            f.write(str(output))
        elif isinstance(output, list):
            f.write(' '.join(map(str, output)))
        elif isinstance(output, tuple):
            f.write(' '.join(output))

def max_same_height(children):
    heights = {}
    for _, height in children:
        if height in heights:
            heights[height] += 1
        else:
            heights[height] = 1
    return max(heights.values())

def arrange_children(children, max_height_count):
    sorted_children = sorted(children, key=lambda x: x[1])
    ascending = sorted_children[:max_height_count]
    descending = sorted_children[max_height_count:]
    descending.reverse()
    arranged_children = [child[1] for child in ascending] + [child[1] for child in descending]
    return arranged_children

def main():
    input_file = "serbare.in"
    output_file = "serbare.out"
    request, children = read_input(input_file)

    if request == 1:
        max_same = max_same_height(children)
        write_output(output_file, max_same)
    elif request == 2:
        max_same = max_same_height(children)
        arranged_children = arrange_children(children, max_same)
        write_output(output_file, arranged_children)

if __name__ == "__main__":
    main()

</syntaxhighlight>