1351 - Nano

În lumea lui Nano totul se construiește la nivel atomic. Știința a ajuns așa departe încât poate construi ”plăci” dreptunghiulare de atomi în care aceștia sunt aliniați perfect, pe un singur strat, formând un rastru. Nano dorește să comande la o firmă plăci pătrate de dimensiuni mari. Dimensiunile sunt atât de mari încât numărul de atomi dintr-o placă poate să fie scris cu până la 500 cifre. Firma i-a dat o listă cu bucățile de material de care dispune, pentru fiecare bucată fiind cunoscut numărul de atomi componenți, urmând ca Nano să aleagă doar acele bucăți din care se pot construi plăci pătrate.

Cerința

Scrieți un program care citind numărul de atomi ai fiecărei bucăți de material din fișierul nanoin.txt scrie în fișierul nanoout.txt doar bucățile de material din care se pot face plăcile dorite de Nano.

Date de intrare

Pe prima linie a fișierului nanoin.txt se află numărul natural n ce reprezintă numărul de bucăți de material, iar pe următoarele 2*n rânduri perechi de numere x y: x reprezentând numărul de cifre a lui y, y fiind numărul de atomi dintr-o bucată de material.

Date de ieșire

Se vor scrie în fișierul nanoout.txt acele valori y din care se pot construi plăci pătrate.

Restricții și precizări

• 1 < n < 50
• 0 < x < 501
• În fișier se află cel puțin o bucată de material ce respectă cerințele lui Nano

Exemplul 1

nanoin.txt

3

2

39

3

100

11

15241383936

nanoout.txt

Datele introduse corespund restricțiilor impuse.

100

15241383936

Explicație

În fişierul nanoin.txt există 3 numere, primul având două cifre, al doilea trei cifre şi ultimul 11 cifre. Din bucata cu 100 atomi se poate face o placă pătrată de latură 10 iar din bucata de dimensiune 15241383936 se obține o placă de latură 123456.

Exemplul 2

nanoin.txt

51

1

2

1

3

nanoout.txt

Datele introduse nu corespund restricțiilor impuse.

Rezolvare

# 1351 - Nano
import math


def este_patrat_perfect(n):
    radacina = math.isqrt(n)
    return radacina * radacina == n


def validare(numere_validare, fisier_iesire_validare):           # functia de validare a datelor de intrare
    if len(numere_validare) > 50:
        fisier_iesire_validare.write("Datele introduse nu corespund restrictiilor impuse\n")
        exit(0)
    for numar in numere_validare:
        if len(str(numar)) > 500:    # fiecare numar trebuie sa aiba mai putin de 500 de cifre
            fisier_iesire_validare.write("Datele introduse nu corespund restrictiilor impuse\n")
            exit(0)
    fisier_iesire_validare.write("Datele introduse corespund restrictiilor impuse\n")


def scrie_bucati_patrate(numere_scriere, fisier_iesire_scriere):
    for numar in numere_scriere:
        if este_patrat_perfect(numar):
            fisier_iesire_scriere.write(str(numar) + '\n')


def citeste_date_intrare():
    with open('nanoin.txt', 'r') as fisier_intrare_citire:
        n_citire = int(fisier_intrare_citire.readline())
        numere_citire = []
        for _ in range(n_citire):
            fisier_intrare_citire.readline()  # citim si ignoram numarul de cifre
            linie = fisier_intrare_citire.readline()
            if not linie:  # daca am ajuns la sfarsitul fisierului sau exista o linie goala, iesim din bucla
                break
            numere_citire.append(int(linie))  # citim numarul
    return n_citire, numere_citire


if __name__ == '__main__':
    n_main, numere_main = citeste_date_intrare()
    with open('nanoout.txt', 'w') as fisier_iesire_main:
        validare(numere_main, fisier_iesire_main)  # validam datele de intrare
        scrie_bucati_patrate(numere_main, fisier_iesire_main)