Editing 4100 - Bipatrat (section)

== Rezolvare == 
<syntaxhighlight lang="python" line>
# 4100

import math

def este_bipatrat(n):
    # Verificăm dacă n este pătrat perfect
    if int(math.sqrt(n))**2 != n:
        return False

    # Obținem lista de numere obținute prin inserarea unei cifre în n
    nums = []
    str_n = str(n)
    for i in range(len(str_n)+1):
        for j in range(10):
            if i == 0 and j == 0:
                continue
            new_num = int(str_n[:i] + str(j) + str_n[i:])
            if int(math.sqrt(new_num))**2 == new_num:
                nums.append(new_num)

    # Verificăm dacă numerele obținute sunt și ele pătrate perfecte
    for num in nums:
        if int(math.sqrt(num))**2 != num:
            return False

    return True

def patrat_perfect(n):
    # Obținem lista de numere obținute prin inserarea unei cifre în n
    nums = []
    str_n = str(n)
    for i in range(len(str_n)+1):
        for j in range(10):
            if i == 0 and j == 0:
                continue
            new_num = int(str_n[:i] + str(j) + str_n[i:])
            if int(math.sqrt(new_num))**2 == new_num:
                nums.append(new_num)

    # Obținem lista de pătrate perfecte din nums
    perfect_squares = []
    for num in nums:
        if int(math.sqrt(num))**2 == num:
            perfect_squares.append(num)

    # Returnăm lista sortată în ordine crescătoare
    return sorted(perfect_squares)

if __name__ == "__main__":
    # Deschidem fișierele de intrare și de ieșire
    with open("bipatrat.in", "r") as fin, open("bipatrat.out", "w") as fout:
        # Citim numărul de linii din fișierul de intrare
        n = int(fin.readline())

        # Parcurgem liniile din fișierul de intrare
        for i in range(n):
            # Citim numărul de pe linia curentă
            num = int(fin.readline())

            # Verificăm dacă numărul este bipătrat
            if este_bipatrat(num):
                # Obținem lista de pătrate perfecte
                perfect_squares = patrat_perfect(num)

                # Scriem lista în fișierul de ieșire
                for square in perfect_squares:
                    fout.write(str(square) + " ")
                fout.write("\n")
            else:
                fout.write("0\n")









</syntaxhighlight>