Editing 2905 - Divizori 4 (section)

== Rezolvare ==
<syntaxhighlight lang="python" line>
# 2905 - Divizori 4
def validare(serie_validare):
    # Verificam daca numarul de elemente din serie este mai mare decat 50_000
    # Daca este, aruncam o exceptie ValueError
    if len(serie_validare) > 50_000:
        raise ValueError

    for numar in serie_validare:
        if not (1 <= numar <= 1_000_000_000):
            raise ValueError

    # Daca datele de intrare sunt valide, afisam un mesaj in fisierul de iesire
    fisier_iesire.write("Datele de intrare corespund restrictiilor impuse\n")


def numar_divizori(n_divizori):
    i = 1
    rezultat = 0
    # Parcurgem numerele de la 1 la radacina patrata a numarului
    while i * i <= n_divizori:
        # Daca numarul nu este divizor, incrementam i
        if n_divizori % i:
            i += 1
            continue
        # Daca i este divizor, il adaugam la rezultat
        # Daca i * i este egal cu numarul, adaugam doar o data i la rezultat
        # Altfel, adaugam si i si numarul / i la rezultat
        if i * i == n_divizori:
            rezultat += 1
        else:
            rezultat += 2
        i += 1
    return rezultat


def divizori4(serie_divizori4, pozitie_divizori4, cerinta_divizori4):
    # Luam numarul de pe pozitia data
    numar = serie_divizori4[pozitie_divizori4-1]
    # Calculam numarul de divizori ai numarului
    divizori_numar = numar_divizori(numar)

    # Daca cerinta este 1, afisam numarul de divizori al numarului
    if cerinta_divizori4 == 1:
        fisier_iesire.write(str(divizori_numar))
    else:
        # Daca cerinta este 2, afisam numerele din serie care au acelasi numar de divizori ca numarul de pe pozitia data
        numere_cu_acelasi_nr_divizori = [x for x in serie_divizori4 if numar_divizori(x) == divizori_numar]
        # Sortam numerele in ordine descrescatoare
        numere_cu_acelasi_nr_divizori.sort(reverse=True)
        # Afisam numerele
        fisier_iesire.write(' '.join(map(str, numere_cu_acelasi_nr_divizori)))


if __name__ == '__main__':
    # Deschidem fisierele de intrare si iesire
    fisier_intrare = open("divizori4in.txt", "r")
    fisier_iesire = open("divizori4out.txt", "w")

    try:
        n_main, cerinta_main = map(int, fisier_intrare.readline().split())
        serie_main = list(map(int, fisier_intrare.readline().split()))
        pozitie_main = int(fisier_intrare.readline())

        # Validam datele de intrare
        validare(serie_main)
        # Rezolvam problema
        divizori4(serie_main, pozitie_main, cerinta_main)

    # Tratam exceptiile care pot aparea
    except ValueError:
        fisier_iesire.write("Datele de intrare nu corespund restrictiilor impuse")
    except IndexError:
        fisier_iesire.write("Datele de intrare nu corespund restrictiilor impuse")

</syntaxhighlight>