2099 - Unu Patru

Sursa: [1]

Enunt

Fiind dat un număr natural, efectuând suma pătratelor cifrelor numărului dat, apoi repetând însumarea pătratelor cifrelor pentru numerele obţinute ca rezultat, la un moment dat se obţine una dintre valorile 1 sau 4. De exemplu, pentru numărul natural 89, se obţin, pe rând, valorile 145, 42, 20, 4 iar pentru numărul natural 86, valorile rezultate sunt 100, 1.

Cerința

Dat un set de numere naturale, să se determine pentru fiecare dintre ele, numărul de repetări ale calculului sumei pătratelor cifrelor până la obţinerea rezultatului 1 sau 4.

Date de intrare

Fişierul de intrare unupatru.in conţine pe prima linie o valoare naturală n, care reprezintă numărul de numere naturale care vor fi testate. Pe linia a doua a fişierului de intrare se găsesc cele n numere naturale, separate prin câte un spaţiu.

Date de ieșire

Fişierul de ieşire unupatru.out conţine pe prima linie n numere naturale, care indică, în ordinea din fişierul de intrare, numărul de repetări ale calculului pentru fiecare dintre cele n numere din fişierul de intrare.

Restricții și precizări

1 <= n <= 10 1 <= numerele testate <= 1018

Exemplul 1:

unupatru.in

2 89 68 unupatru.out

4 2

Explicație

Se obţin, pe rând, pentru 89, valorile 145 42 20 4 Se obţin, pe rând, pentru 68, valorile 100 1

Exemplul 2:

unupatru.in

3 33 17 638 unupatru.out

10 9 5

Explicație

Valorile obţinute pentru cele 3 valori sunt: 18 65 61 37 58 89 145 42 20 4 50 25 29 85 89 145 42 20 4 109 82 68 100 1

Încărcare soluție

def sum_of_squares(n):
    # Calculează suma pătratelor cifrelor numărului n
    return sum(int(digit)**2 for digit in str(n))

def count_iterations(n):
    # Calculează numărul de repetări necesare pentru a ajunge la 1 sau 4
    if n == 1 or n == 4:
        return 1
    else:
        return 1 + count_iterations(sum_of_squares(n))

# Citim numărul de teste
with open("unupatru.in", "r") as f:
    n = int(f.readline())

# Citim numerele din fișier și le prelucrăm
with open("unupatru.in", "r") as f:
    numbers = list(map(int, f.readline().split()))

# Calculăm numărul de repetări pentru fiecare număr și scriem rezultatele în fișier
with open("unupatru.out", "w") as f:
    for number in numbers:
        f.write(str(count_iterations(number)) + " ")