0132 - ElemPP

Cerinţa

Se citeşte n şi un şir de n numere naturale. Să se afișeze elementele din șir care au proprietatea că primele două cifre din scrierea zecimală formează un pătrat perfect.

Date de intrare

Fişierul de intrare elempp.in conţine pe prima linie numărul n şi pe a doua linie n numere naturale separate prin spaţii.

Date de ieșire

Dacă datele sunt introduse corect, pe ecran se va afișa: "Datele sunt introduse corect.", apoi, fişierul de ieşire elempp.out va conţine pe prima linie elementele din șirul dat care respectă proprietatea precizată (vezi cerință). Ordinea acestor elemente este cea în care se află în șirul dat. În cazul în care datele nu respectă restricțiile, se va afișa pe ecran: "Datele nu corespund restricțiilor impuse."'.

Restricţii şi precizări

1 ⩽ n ⩽ 20
numerele de pe a doua linie a fişierului de intrare vor fi mai mici decât 1.000.000.000 și au cel puțin două cifre

Exemple

Exemplul 1

elempp.in: 5; 253 16 225 100 3678
Ecran: Datele sunt introduse corect.
elempp.out: 253 16 3678

Exemplul 2

elempp.in: 5; 5 3 2 4 1
Ecran: Datele nu corespund restricțiilor impuse.

Rezolvare

0132 ElemPP

import math

def este_patrat_perfect(n):

   radical_n = int(math.sqrt(n))
   return radical_n * radical_n == n

def afisare_elem_primele_2_cifre_pp(vector):

   rezultat = []
   for numar in vector:
       if este_patrat_perfect(int(str(numar)[:2])):
           rezultat.append(str(numar))
   with open('elempp.out', 'w') as fisier_iesire:
       fisier_iesire.write(' '.join(rezultat))

def citire_conform_restrictiilor(n, vector):

   if n < 1 or n >= 20:
       print("Datele nu corespund restricțiilor impuse.")
       exit()
   for element in vector:
       if element < 10 or element >= 1000000000:
           print("Datele nu corespund restricțiilor impuse.")
           exit()
   if n != len(vector):
       print("Datele nu corespund restricțiilor impuse.")
       exit()
   print("Datele sunt introduse corect.")

if __name__ == '__main__':

   with open('elempp.in', 'r') as fisier_intrare:
       n = int(fisier_intrare.readline().strip())
       vector = list(map(int, fisier_intrare.readline().strip().split()))
   citire_conform_restrictiilor(n, vector)
   afisare_elem_primele_2_cifre_pp(vector)

</syntaxhighlight>

Explicație rezolvare

   Programul de mai sus conține trei funcții, funcția este_patrat_perfect(n), funcția afisare_elem_primele_2_cifre_pp(vector) și funcția citire_conform_restrictiilor(n, vector), care se vor rula în interiorul main-ului (if __name__ == '__main__' , linia 34) după citirea numărului n (linia 36) și celor n numere pe care le vom pune în șirul „vector” (linia 37), pe care le citim prin deschiderea fișierului elempp.in (linia 35).
   După ce am citit toate datele de intrare, se va apela funcția citire_conform_restrictiilor(n, vector) care primește doi parametri: n și vector. Funcția verifică n între 1 și 20 (linia 21), dacă elementele vectorului „vector” sunt mai mici decât 1000000000, respectiv strict mai mari decât 10 (liniile 24-25, trebuie să fie strict mai mari decât 10 deoarece numerele trebuie să aibă cel puțin două cifre) și dacă n este lungimea vectorului „vector” (linia 28). Dacă oricare dintre condiții este încălcată, se va afișa pe ecran mesajul „Datele nu corespund restricțiilor impuse.” și se va ieși din program cu comanda exit(). Dacă toate condițiile sunt respectate, se va afișa mesajul „Datele sunt introduse corect.” (linia 31) și se va continua programul.
   Dacă s-au introdus corect datele, se va apela funcția afisare_elem_primele_2_cifre_pp(vector) care primește ca parametrii numărul n și vectorul „vector”, apoi:
-> se inițializează o listă vidă rezultat care va conține elementele care respectă proprietatea cerută;
-> parcurge fiecare număr din vector utilizând un ciclu for (linia 13);
-> pentru fiecare număr numar din vector, extrage primele două cifre din scrierea zecimală a acestuia utilizând str(numar)[:2], care convertește numar într-un șir de caractere și apoi obține primele două caractere din acest șir;
-> convertește cele două caractere extrase într-un număr întreg utilizând int(str(numar)[:2]);
-> verifică dacă acest număr întreg este un pătrat perfect utilizând funcția este_patrat_perfect(n), care determină dacă radicalul pătrat al lui n (calculat cu math.sqrt(n), este necesar la începutul programului să importăm biblioteca math, linia 3) este un număr întreg. Dacă este un pătrat perfect, adaugă numar în lista rezultat;
-> deschide fișierul de ieșire 'elempp.out' în modul de scriere ('w');
-> scrie elementele din lista rezultat separate prin spații în fișierul de ieșire utilizând metoda write() a obiectului fișier (fisier_iesire.write(' '.join(rezultat))), astfel încât ele să fie afișate în aceeași linie în fișierul de ieșire.
   Astfel, funcția afisare_elem_primele_2_cifre_pp(vector) identifică și scrie în fișierul de ieșire elementele din vectorul dat care au primele două cifre din scrierea zecimală care formează un pătrat perfect, în ordinea în care se află în vectorul de intrare.