Vizualizare sursă pentru „3459 - Count Prime”

== Cerinţa ==
Să se calculeze cate numere prime sunt în intervalul '''[stanga, dreapta]'''.
== Date de intrare ==
Fișierul de intrare '''countprime.in''' conține pe prima linie două numere '''stanga''' și '''dreapta'''.
== Date de ieşire ==
Dacă datele sunt introduse corect, în fișier se va afișa:'''"Datele sunt introduse corect."''', apoi pe un rând nou fișierul de ieșire '''countprime.out''' va conține pe prima linie numărul '''cnt''', reprezentând numărul de numere prime din intervalul dat. În cazul în care datele nu respectă restricțiile, se va afișa în fișier:'''"Datele nu corespund restricțiilor impuse."'''.
== Restricții și precizări ==
* 1 &les; stanga &les; dreapta &les; 2 ** 32
* dreapta - stanga &les; 1.000.000
== Exemplu ==
; countprime.in
: 2 10
; countprime.out
: Datele sunt introduse corect.

== Explicație == 
Sunt '''4''' numere prime în intervalul '''[2, 10]'''.
== Rezolvare ==
<syntaxhighlight lang="python" line>

def validare_date(stanga, dreapta):
    flag = False
    if stanga.isdigit() and dreapta.isdigit(): #verificam daca ambele string-uri pot fi convertite in numere intregi
        if 0 < int(stanga) < int(dreapta) <= 2 ** 32 and int(dreapta) - int(stanga) <= 1_000_000: #verificam daca cele doua numere se afla in intervalul si au o diferenta mai mica sau egala cu 1.000.000
            flag = True
    return flag

def prim(n):
    #Funcția verifică dacă un număr este prim
    if n < 2:
        return False
    for i in range(2, int(n ** 0.5) + 1):
        if n % i == 0:
            return False
    return True

def câte_prime(stanga, dreapta):
    #Funcția calculează numărul de numere prime din intervalul [stanga, dreapta]
    nr = 0
    for n in range(stanga, dreapta + 1):
        if prim(n):
            nr += 1
    return nr

if __name__ == '__main__':
    with open('countprime.in', 'r') as fin, open('countprime.out', 'w') as fout:
        stanga, dreapta = fin.readline().split() #citim cele doua numere din fișierul de intrare

        if validare_date(stanga, dreapta): #verificăm dacă cele două numere respectă condițiile impuse
            cnt = câte_prime(int(stanga), int(dreapta)) #calculăm numărul de numere prime din intervalul [stanga, dreapta]

            fout.write("Datele sunt introduse corect.\n") #scriem mesajul in fișierul de ieșire
            fout.write(str(cnt)) #scriem in fisierul de ieșire numărul de numere prime

        else:
            fout.write("Datele introduse nu  corespund restrictiilor impuse.")#în cazul în care nu sunt respectate condițiile impuse afișăm mesajul corespunzător

</syntaxhighlight>