Editing 2880 - Gradinita

== Cerinta ==

La grădinița Prichindel sunt copii cuminți și copii obraznici. Moș Crăciun a adus cadouri doar pentru câțiva copii cuminți. Problema este că nu se știe exact care copil e obraznic și care este cuminte. Ajutați-o pe doamna educatoare să afle care copil primește cadou și care nu. Se cunoaște faptul că la grădiniță sunt exact n copii și fiecare copil are inscripționat pe uniformă un număr natural nenul, distinct.

În prima etapă, sunt scoși din lista Moșului copiii obraznici care sunt cei ce au inscripționate pe uniformă numere prime. În a două etapă, din copiii rămași, sunt scoși și cei care au suma cifrelor de pe pozițiile impare egală cu un număr prim (cifrele se numerotează de la stânga la dreapta, începând cu poziția 1, corespunzătoare cifrei cea mai semnificative).

În a treia etapă (ultima), se  elimină din lista copiilor cuminți rămași după primele două etape, din k în k până când se ajunge la capătul listei.

Să se determine copiii care au rămas pe lista copiilor cuminți după toate cele trei etape eliminatorii, sau mesajul Toti copiii sunt obraznici!, dacă toți copiii au fost eliminați.

== Date de intrare ==

Pe prima linie a fișierului de intrare gradinita.txt se află două numere n și k cu semnificația din enunț.  Pe a două linie a fișierului de intrare se află n numere naturale, cu semnificația din enunț.

== Date de iesire ==

În fișierul de ieșire gradinita.txt se va afișa lista după a treia etapă eliminatorie conform valorii cerinței.

== Restrictii si precizari ==

*1 ≤ n ≤ 100000
*1 ≤ x ≤ 100000 (unde x reprezintă numărul inscripționat pe uniforma unui copil)

== Exemplul 1 ==
;Intrare
:9 4
:3 102 31 333 46 1022 10 4848 21 

;Iesire
;Datele introduse corespund restrictiilor impuse
:333 46 10  

== Exemplul 2 ==
;Intrare
:-5 7
:5 309 -57 982 4
;Iesire
;Datele introduse nu corespund restrictiilor impuse


== 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_impare(numar):
    suma = 0
    for i in range(len(str(numar))):
        if (i + 1) % 2 == 1:  # verificăm dacă poziția cifrei este impară
            suma += int(str(numar)[i])
    return suma

def rezolva_problema(n, k, copii):
    # Etapa 1: Eliminarea copiilor cu numere prime
    copii_obraznici = [copil for copil in copii if not este_prim(copil)]

    # Etapa 2: Eliminarea copiilor cu suma cifrelor impare număr prim
    copii_obraznici = [copil for copil in copii_obraznici if not este_prim(suma_cifre_impare(copil))]

    # Etapa 3: Eliminarea copiilor cuminți din k în k
    copii_cuminți = [copil for i, copil in enumerate(copii) if (i + 1) % k != 0]

    return copii_cuminți

# Citire date de intrare
with open("gradinita.txt", "r") as file:
    n, k = map(int, file.readline().split())
    copii = list(map(int, file.readline().split()))

# Rezolvare problema
rezultat = rezolva_problema(n, k, copii)

# Scriere rezultat în fișierul de ieșire
with open("gradinita.txt", "w") as file_out:
    if len(rezultat) > 0:
        file_out.write(" ".join(map(str, rezultat)))
    else:
        file_out.write("Toti copiii sunt obraznici!")
</syntaxhighlight>