Editing 3763 - Puternic (section)

== Rezolvare == 
<syntaxhighlight lang="python" line="1">
# 3763 - Puternic
import math


# Funcția de validare a datelor de intrare
def valideaza_intrare(C: int, N: int, nums: list[int]) -> bool:
    # Verificăm că C este 1 sau 2
    if C not in [1, 2]:
        return False

    # Verificăm că N este între 1 și 100.000
    if not (1 <= N <= 100000):
        return False

    # Verificăm că lista nums are lungimea N și conține doar numere întregi între 1 și 1.000.000.000
    if len(nums) != N or not all(isinstance(x, int) and 1 <= x <= 1000000000 for x in nums):
        return False

    return True


# Funcția de rezolvare
def rezolva(C: int, N: int, nums: list[int]) -> str:
    # Cazul 1
    if C == 1:
        nrp = 0
        for x in nums:
            if x > 1 and este_puternic(x):
                nrp += 1
        return str(nrp)

    # Cazul 2
    nums_noi = []
    for x in nums:
        if x == 1 or not este_puternic(x):
            nums_noi.append(x)

    result = []
    for i in range(len(nums_noi) // 2):
        nr1 = nums_noi[i]
        nr2 = nums_noi[-i - 1]
        lp = concateneaza(nr1, nr2)
        if este_puternic(lp):
            result.append((nr1, nr2))

    if not result:
        return "-1"
    else:
        return "\n".join([f"{nr1} {nr2}" for nr1, nr2 in result])


# Funcția de verificare dacă un număr este puternic
def este_puternic(nr: int) -> bool:
    # Calculăm descompunerea în factori primi și verificăm dacă toți au exponența mai mare sau egală cu 2
    i = 2
    while i * i <= nr:
        ex = 0
        while nr % i == 0:
            ex += 1
            nr //= i
        if ex == 1:
            return False
        i += 1
    if nr > 1:
        ex = 1
    if ex == 1:
        return False

    # Verificăm dacă nr este pătrat perfect sau cub perfect
    sqrt_nr = round(math.sqrt(nr))
    if sqrt_nr * sqrt_nr == nr:
        return True
    cub_rt = round(nr ** (1 / 3))
    if cub_rt * cub_rt * cub_rt == nr:
        return True
    return False


# Funcția de concatenare a două numere întregi
def concateneaza(a: int, b: int) -> int:
    s = str(a) + str(b)
    return int(s)

if __name__ == "__main__":
    with open("puternic.in", "r") as fin:
        with open("puternic.out", "w") as fout:
            C = int(fin.readline().strip())
            N = int(fin.readline().strip())
            nums = list(map(int, fin.readline().strip().split()))

            if not valideaza_intrare(C, N, nums):
                 print("Datele nu corespund restricțiilor impuse.")
            else:
                print("Datele sunt introduse corect.\n")
                rezultat = rezolva(C, N, nums)
                fout.write(rezultat)
               










</syntaxhighlight>