Editing 1460 - Serbare (section)

== Rezolvare ==
<syntaxhighlight lang="python" line>
# 1460  Serbare
def validare(n_validare, p_validare, grupe_validare):
    if n_validare < 1 or n_validare > 1000:    # Verificăm dacă n este în intervalul 1-1000
        raise ValueError  # Ridicăm o eroare dacă n nu este în intervalul 1-1000
    if p_validare < 1 or p_validare > 11:    # Verificăm dacă p este în intervalul 1-11
        raise ValueError  # Ridicăm o eroare dacă p nu este în intervalul 1-11
    for grup in grupe_validare:    # Parcurgem lista de grupe
        # Verificăm dacă numărul de copii este în intervalul 1-100
        if grup[0] < 1 or grup[0] > 100:
            raise ValueError
        # Verificăm dacă tipul de uniformă este în intervalul 1-p
        if grup[1] < 1 or grup[1] > p_validare:
            raise ValueError
    file_out.write("Datele de intrare corespund restrictiilor impuse.\n")


def uniforme(grupe):
    uni = {}  # Inițializăm un dicționar gol
    for grup in grupe:  # Parcurgem lista de grupe
        # Dacă tipul de uniformă nu se află încă în dicționar, îl adăugăm cu valoarea 0
        if grup[1] not in uni:
            uni[grup[1]] = 0
        # Adăugăm numărul de copii la totalul pentru tipul respectiv de uniformă
        uni[grup[1]] += grup[0]
    # Sortăm tipurile de uniforme în ordinea descrescătoare a numărului total de copii
    sort_uni = sorted(uni.items(), key=lambda x: x[1], reverse=True)
    return [uni[0] for uni in sort_uni]    # Returnăm tipurile de uniforme sortate


if __name__ == '__main__':
    file_in = open("serbarein.txt", "r")
    file_out = open("serbareout.txt", "w")

    try:
        # Citim numărul de grupe și tipuri de uniforme
        n_main, p_main = map(int, file_in.readline().split())
        # Citim datele pentru fiecare grup
        grupe_main = [list(map(int, file_in.readline().split())) for _ in range(n_main)]
        # Validăm datele de intrare
        validare(n_main, p_main, grupe_main)
        # Calculăm numărul total de copii pentru fiecare tip de uniformă
        rezultat_main = uniforme(grupe_main)
        # Scriem tipurile de uniforme în fișierul de ieșire, separate printr-un spațiu
        file_out.write(' '.join(str(uni) for uni in rezultat_main) + '\n')

    except ValueError:
        file_out.write("Datele de intrare nu corespund restrictiilor impuse.")
    except IndexError:
        file_out.write("Datele de intrare nu corespund restrictiilor impuse.")

</syntaxhighlight>