2601 - Sumă Pătrate Cifre

Cerința

Scrieți un program care calculează suma pătratelor cifrelor unui număr natural de trei cifre citit de la tastatură.

Date de intrare

Programul citește de la tastatură numărul num.

Date de ieșire

Pe ecran se va afișa mesajul: "Datele de intrare corespund restricțiilor impuse". În următorul rând se va afișa pe ecran rezultatul cerut.

În cazul în care datele introduse de la tastatură nu îndeplinesc cerințele enunțate, pe ecran se va afișa mesajul "Datele de intrare nu corespund restricțiilor impuse".

Restricții și precizări

100 ⩽ num ⩽ 999

Exemplu 1

Intrare: 221
Ieșire: Datele de intrare corespund restricțiilor impuse; 9

Exemplu 2

Intrare: 1000
Ieșire: Nu au fost respectate cerintele impuse

Rezolvare

2601 - sumapatratecifre

def validare(num):

   # Daca datele sunt invalide v-om face raise la eroarea ValueError

   if not (num >= 100 and num <= 999):
       raise ValueError

def sumapatratecifre(num):

   # Convertim num in string dupa care generam o lista in care
   # avem fiecare caracter lui num care se converteste in int si se pune
   # la puterea a doua
   # Lista o punem in functia sum care va aduna toate cifrele din lista
   suma = sum([int(cifra) ** 2 for cifra in str(num)])

   print(suma)

if __name__ == '__main__':

   # Citim de la tastatura numerele cerute
   num = int(input())

   # Verificam daca datele introduse sunt valide
   # daca nu afisam mesajul corespunsador
   try:
       validare(num)
   except ValueError:
       print("Nu au fost respectate cerintele impuse")
   else:
       print("Datele de intrare corespund restricțiilor impuse")

       sumapatratecifre(num)

</syntaxhighlight>