Editing 2384 - Divigrup (section)

== Rezolvare ==
<syntaxhighlight lang="python" line>
# 2384 - Divigrup
import math


def validare(numar_elemente1, lista_numere1):           # functia de validare a datelor de intrare
    if numar_elemente1 < 1 or numar_elemente1 > 200:
        raise ValueError
    for numar in lista_numere1:
        if numar < 1 or numar > 1000000000:
            raise ValueError


def calculeaza_numar_divizori(numar):          # functia numar_divizori calculează numărul de divizori ai unui număr.
    numar_divizori1 = 0
    for i in range(1, int(math.sqrt(numar)) + 1):
        if numar % i == 0:
            if numar / i == i:
                numar_divizori1 += 1
            else:
                numar_divizori1 += 2
    return numar_divizori1


def divigrupuri(lista_numere1):  # functia divigrupuri grupează numerele din șir în funcție de numărul lor de divizori.
    grupuri_divizori1 = {}
    for numar in lista_numere1:
        numar_divizori1 = calculeaza_numar_divizori(numar)
        if numar_divizori1 not in grupuri_divizori1:
            grupuri_divizori1[numar_divizori1] = []
        grupuri_divizori1[numar_divizori1].append(numar)
    for numar_divizori1 in grupuri_divizori1:
        grupuri_divizori1[numar_divizori1].sort()
    return grupuri_divizori1


if __name__ == '__main__':
    file_in = open("divigrupin.txt", "r")         # declararea fisierelor
    file_out = open("divigrupout.txt", "w")       # fisierul out trebuie declarat cu optiunea "w" (write)

    try:
        numar_elemente = int(file_in.readline())   # citirea numărului se face ca sir de caractere deoarece e prea mare
        lista_numere = list(map(int, file_in.readline().split()))

        validare(numar_elemente, lista_numere)                 # apelul functiei de validare
        file_out.write("Datele de intrare corespund restrictiilor impuse\n")
        grupuri_divizori = divigrupuri(lista_numere)  # apelul functiei de rezolvare

        file_out.write(str(len(grupuri_divizori)) + "\n")
        for numar_divizori in sorted(grupuri_divizori.keys(), reverse=True):
            file_out.write(str(len(grupuri_divizori[numar_divizori])) + " ")
            file_out.write(" ".join(map(str, grupuri_divizori[numar_divizori])) + "\n")

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

</syntaxhighlight>