Vizualizare sursă pentru „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 "Nu au fost respectate cerintele impuse".
== Restricții și precizări ==
* 100 &les; '''num''' &les; 999
== Exemplu 1 ==
; Intrare
: 221
; Ieșire
: Datele de intrare corespund restricțiilor impuse
: 9
<br>
== Exemplu 2 ==
; Intrare
: 1000
; Ieșire
: Nu au fost respectate cerintele impuse
<br>
== Rezolvare ==
<syntaxhighlight lang="python" line>
# 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>