Editing 2129 - Prime 1 (section)

==Rezolvare==
<syntaxhighlight lang="python" line="1">
#2129 Prime 1
def cerinte(c, n, numere):
    if not n == len(numere):
        return False
    if not 1 < n <= 50:
        return False

    if c == 1 or c == 3:
        for nr in numere:
            if not 1 < nr < 10**7:
                return False

    return True


def prim(nr):
    # Dace nr <= 1, atunci nr nu este prim.
    if nr < 2:
        return False
    # Dacă de la 2 până la rădăcina pătrată a lui nr există un divizor a lui nr, atunci nr nu este prim.
    for i in range(2, int(nr**0.5) + 1):
        if nr % i == 0:
            return False

    return True


def suma_cifre(nr):
    # Transformăm numărul într-o listă de cifre, după care returnăm suma cifrelor
    cifre = [int(x) for x in list(str(nr))]
    return sum(cifre)


def suma_factori_cifre(nr):
    suma = 0
    # Primul factor prim este 2
    factor = 2

    while nr > 1:
        exponent = 0

        # Dacă numărul este divizibil cu factorul curent...
        if nr % factor == 0:
            # ...îl împărțim la factorul curent și numărăm de câte ori apare factorul curent în număr
            while nr % factor == 0:
                nr //= factor
                exponent += 1
            # Adăugăm la sumă suma cifrelor factorului și a exponentului, dacă exponentul este mai mare decât 1
            suma += suma_cifre(factor)
            if exponent > 1:
                suma += suma_cifre(exponent)
        # Trecem la următorul factor prim
        factor += 1

        # Dacă factorul prim la pătrat este mai mare decât numărul, atunci numărul este prim
        if factor ** 2 > nr:
            factor = nr

    return suma


def prime_suma(nr):
    # Dacă nr este par sau nr-2 este prim, atunci nr nu este sumă de două numere prime.
    if nr % 2 == 0:
        return False
    elif prim(nr - 2):
        return False

    return True


def c_prim_fibonacii(numere):
    nr = 0
    fibonacii = {0}
    a, b = 1, 1

    # Generăm numere Fibonacii până ajungem la 10^7
    while a < 10 ** 7:
        # Algoritm pentru a genera numerele din șirul lui Fibonacci
        fibonacii.add(a)
        temp = a
        a = b
        b += temp

    # Pentru fiecare număr din numere...
    for numar in numere:
        # ...dacă numărul se află în șirul lui Fibonacci și este prim...
        if numar in fibonacii and prim(numar):
            # ...îl numărăm
            nr += 1

    return nr


def c_economice(numere):
    nr = 0
    for numar in numere:
        nr_suma_cifre = suma_cifre(numar)
        nde = suma_factori_cifre(numar)

        if nr_suma_cifre > nde:
            nr += 1

    return nr


def c_not_suma(numere):
    nr = 0
    for numar in numere:
        if prime_suma(numar):
            nr += 1

    return nr


def prime1(c, numere):
    # c este numărul cerinței
    if c == 1:
        print(c_prim_fibonacii(numere))

    elif c == 2:
        print(c_economice(numere))

    elif c == 3:
        print(c_not_suma(numere))


if __name__ == "__main__":
    c = int(input())
    n = int(input())
    numere = [int(x) for x in input().split()]

    if not cerinte(c, n, numere):
        print("Datele de intrare nu corespund restricțiilor impuse.")
    else:
        print("Datele de intrare corespund restricțiilor impuse.")
        prime1(c, numere)
</syntaxhighlight>