happy_number_set.py

class Solution:
    def sum_of_squares(self, n: int) -> int:
        total = 0
        while n > 0:
            total += (n % 10) ** 2
            n = int(n / 10)
        return total
    
    def isHappy(self, n: int) -> bool:
        repeat_numbers = set()
        while n != 1:
            total = self.sum_of_squares(n)
            if total in repeat_numbers:
                return False
            else:
                repeat_numbers.add(total)
            n = total
        return True