Vizualizare sursă pentru „2328 - prim013”

Sursa: [https://www.pbinfo.ro/probleme/2328/prim013 - prim013]
----
== Cerinţa ==
Dorel este pasionat de feng shui. Astfel, pentru a-şi crea un cadru adecvat meditaţiei, s-a hotărât să scrie pe pereţii casei numere care au un număr prim de divizori. El a găsit n numere, însă nu ştie să le aleagă pe cele potrivite.
Se dau '''n''' numere naturale. Aflaţi câte dintre acestea au un număr prim de divizori.
== Date de intrare ==
Fișierul de intrare '''prim013.in''' conține pe prima linie numărul '''n''', iar pe a doua linie '''n''' numere naturale nenule separate prin spații.
== Date de ieșire == 
Dacă datele sunt introduse corect, pe ecran se va afișa: 
"Datele sunt corecte.", iar apoi in fișierul de ieșire '''prim013.out''' va conține pe prima linie numărul de numere care au un număr prim de divizori. În caz contrar, se va afișa pe ecran: "Datele nu sunt comform restricțiilor impuse.".

== Restricţii şi precizări ==
* 1 ≤ '''n''' ≤ 1.000.000
* numerele de pe a doua linie a fișierului de intrare vor fi mai mici decât '''10.000.000'''
== Exemple ==
===Exemplul 1===
; prim013.in
: 5
: 1 3 6 9 625
; Ieșire
: Datele sunt corecte.
; prim013.out
: 3
===Exemplul 2===
; prim013.in
: 5
: 1 10 12 15 19999
; Ieșire
: Datele sunt corecte.
; prim013.out
: 0
===Exemplul 3===
; prim013.in
: 2 
: 314441 41241241
; Ieșire
: Datele nu sunt comform restricțiilor impuse.
<br>
== Rezolvare == 
<syntaxhighlight lang="python" line>
#2328 prim013

def este_prim(n):
    if n < 2:
        return False
    for i in range(2, int(n ** 0.5) + 1):
        if n % i == 0:
            return False
    return True


def prim013(n,vector):
    numere_cu_divizori_primi = 0
    for num in vector:
        num_divizori = 0
        for i in range(1, num + 1):
            if num % i == 0:
                num_divizori += 1
        if este_prim(num_divizori):
            numere_cu_divizori_primi += 1
    with open('prim013.out','w') as f:
        f.write(str(numere_cu_divizori_primi))
    

def conform_restrictiilor():
    with open('prim013.in') as f:
        n = int(f.readline())
        vector = list(map(int, f.read().split()))
    if n > 1000000:
        print("Datele nu sunt conform restricțiilor impuse.")
        exit()
    for x in vector:
        if x > 10000000:
            print("Datele nu sunt conform restricțiilor impuse.")
            exit()
    print("Datele sunt corecte.")
    return n , vector 


if __name__ == '__main__':
    n , vector= conform_restrictiilor()
    prim013(n , vector)

</syntaxhighlight>

==Explicaţie cod==

Acest cod implementează o funcție numită '''prim013''', care primește un număr întreg '''n''' și un vector de '''n''' numere întregi '''vector''', și își propune să determine numărul de elemente din vector care au un număr prim de divizori.

Funcția '''este_prim''' este o funcție ajutătoare care primește un număr întreg și returnează adevărat dacă numărul este prim, iar fals în caz contrar.

Funcția '''conform_restrictiilor''' citește datele de intrare din fișierul '''prim013.in''' și verifică dacă acestea respectă restricțiile impuse. În cazul în care datele nu sunt conforme, programul se încheie prin apelarea funcției '''exit()''', iar în cazul în care datele sunt conforme, funcția returnează cele două variabile '''n''' și '''vector'''.

Funcția '''prim013''' parcurge fiecare element din '''vector''' și numără numărul de divizori ai fiecărui element. Dacă numărul de divizori este prim, acesta este adăugat la numărul total de elemente cu număr prim de divizori. Rezultatul este scris în fișierul '''prim013.out'''.