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 ⩽ 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)