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 gradinitain.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 gradinitaout.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

gradinitain.txt

9 4

3 102 31 333 46 1022 10 4848 21

gradinitaout.txt

Datele introduse corespund restrictiilor impuse

333 46 10

Exemplul 2

gradinitain.txt

-5 7

5 309 -57 982 4

Datele introduse nu corespund restrictiilor impuse

Rezolvare

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!')