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

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

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

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()