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

1. 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()))

1. Rezolvare problema

rezultat = rezolva_problema(n, k, copii)

1. 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>