Editing 3733 - Cosuri (section)

== Rezolvare == 
<syntaxhighlight lang="python" line="1">
# 3733 - Cosuri
import math


# Funcția de validare a datelor de intrare
# Funcția de validare a datelor de intrare
def validate_input(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 solve(C: int, N: int, nums: list[int]) -> str:
    # Cazul 1
    if C == 1:
        nrp = 0
        for x in nums:
            if x > 1 and is_powerful(x):
                nrp += 1
        return str(nrp)

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

    result = []
    for i in range(len(new_nums) // 2):
        nr1 = new_nums[i]
        nr2 = new_nums[-i - 1]
        lp = concat(nr1, nr2)
        if is_powerful(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 concatenare și verificare dacă numărul obținut este puternic
def is_powerful(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 concat(a: int, b: int) -> int:
    s = str(a) + str(b)
    return int(s)


# Funcția main care citește și afișează datele din fișiere
# Funcția main care citește și afișează datele din fișiere
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 validate_input(C, N, nums):
                 print ("Datele nu corespund restricțiilor impuse.")
            else:
                print ("Datele sunt introduse corect.\n")
                if C == 1:
                    nrp = 0
                    for x in nums:
                        if x > 1 and is_powerful(x):
                            nrp += 1
                    fout.write(str(nrp))
                elif C == 2:
                    new_nums = []
                    for x in nums:
                        if x == 1 or not is_powerful(x):
                            new_nums.append(x)

                    result = []
                    for i in range(len(new_nums) // 2):
                        nr1 = new_nums[i]
                        nr2 = new_nums[-i - 1]
                        lp = concat(nr1, nr2)
                        if is_powerful(lp):
                            result.append((nr1, nr2))

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










</syntaxhighlight>