Editing 2880 - Gradinita (section)

== Rezolvare ==
<syntaxhighlight lang="python3" line="1">
def este_prim(numar):
    if numar < 2:
        return False
    for i in range(2, int(numar**0.5) + 1):
        if numar % i == 0:
            return False
    return True

def suma_cifre(numar):
    return sum(int(cifra) for cifra in str(numar))

def verificare_date_intrare(n, k, copii):
    if not (1 <= n <= 100000) or not (1 <= k <= 100000):
        return False
    for copil in copii:
        if not (1 <= copil <= 100000):
            return False
    return True

# Citire din fișierul de intrare
with open('gradinitain.txt', 'r') as file:
    n, k = map(int, file.readline().split())
    copii = list(map(int, file.readline().split()))

# Verificare date de intrare
if not verificare_date_intrare(n, k, copii):
    print("Datele introduse nu corespund restrictiilor impuse")
else:
    # Etapa 1: Eliminare copii obraznici cu numere prime
    copii = [copil for copil in copii if not este_prim(copil)]

    # Etapa 2: Eliminare copii obraznici cu suma cifrelor de pe poziții impare primă
    copii = [copil for copil in copii if not este_prim(suma_cifre(copil))]

    # Etapa 3: Eliminare copii cuminți din k în k
    if copii:
        copii = copii[k-1::k]

    # Scriere în fișierul de ieșire
    with open('gradinitaout.txt', 'w') as file:
        if copii:
            file.write(' '.join(map(str, copii)))
        else:
            file.write('Toti copiii sunt obraznici!')


</syntaxhighlight>