0996 - Div 3

Cerință

Pe prima linie a fişierului div3.in se află numărul n, iar pe a doua linie n numere naturale.

Date de intrare

Pe prima linie a fişierului div3.in se află numărul n, iar pe a doua linie n numere naturale.

Date de ieșire

Dacă datele sunt introduse corect, pe ecran se va afișa: "Datele de intrare corespund restricțiilor impuse." Pe prima linie a fişierului div3.out se vor afla elementele şirului dat pentru care suma cifrelor este divizibilă cu 3, separate printr-un spațiu. În caz contrar, se va afișa pe ecran: "Datele de intrare nu corespund restricțiilor impuse."

Restricții de precizări

• 1 ⩽ n ⩽ 1000
• elementele şirului vor avea cel mult 9 cifre

Exemplul 1

Intrare

7

124 51 231 7 24 31 5

Ieșire

Datele introduse corespund restricțiilor impuse.

51 231 24

Exemplul 2

Intrare

5

124 51 abc 7 24

Ieșire

Datele introduse nu corespund restricțiilor impuse.

Rezolvare ver. 1

<syntaxhighlight lang="python" line="1" start="1"> def validare(n, numere):

   # Verifica daca datele de intrare sunt valide conform cerintei problemei.
   # Verifica daca n este un numar intreg pozitiv
   # si daca toate numerele din lista numere sunt numere intregi pozitive.
   return n.isdigit() and all(x.isdigit() for x in numere) and 1 <= int(n) <= 1000

def suma_cifre(n, numere):

   # Calculeaza suma cifrelor pentru fiecare numar din lista numere,
   # si retine doar numerele pentru care suma cifrelor este divizibila cu 3.
   suma_div3 = []
   for x in numere:
       if sum(int(d) for d in str(x)) % 3 == 0:
           suma_div3.append(x)
   return suma_div3

if __name__ == "__main__":

   with open("div3.in", "r") as fin:
       n = fin.readline().strip()  # Citeste numarul de elemente din sir
       numere = fin.readline().strip().split()  # Citeste lista de numere

   if validare(n, numere):
       n = int(n)  # Converteste n la integer
       numere = list(map(int, numere))  # Converteste lista de numere la integers

       print("Datele introduse corespund restricțiilor impuse.")
       suma_div3 = suma_cifre(n, numere)  # Calculeaza suma_div3
       with open("div3.out", "w") as fout:
           fout.write(" ".join(map(str, suma_div3)))  # Scrie numerele in fisierul de iesire
           fout.write("\n")
   else:
       print("Datele introduse nu corespund restricțiilor impuse.")

</syntaxhighlight>