LeetCode / Happy Number

https://leetcode.com/problems/happy-number/

Write an algorithm to determine if a number is "happy".

A happy number is a number defined by the following process: Starting with any positive integer, replace the number by the sum of the squares of its digits, and repeat the process until the number equals 1 (where it will stay), or it loops endlessly in a cycle which does not include 1. Those numbers for which this process ends in 1 are happy numbers.

Example:

Input: 19 Output: true
Explanation:
1² + 9² = 82
8² + 2² = 68
6² + 8² = 100
1² + 0² + 0² = 1

ある値について、各桁の値の二乗和をとり、新しくできた数についても各桁の値の二乗和をとり、、、と繰り返したとき、最終的に1となる数をハッピー数と呼ぶそうです。

与えられた値がハッピー数であるかどうかを判定する問題。易しいです。

解答・解説

解法1

各桁の値の二乗和をとる過程で、1となる前に同じ値が2度出現したらループに入っているため、ハッピー数となりません。

class Solution:
    def isHappy(self, n: int) -> bool:
        seens = set()
        while n != 1:
            n = sum([int(i)**2 for i in str(n)])
            if n in seens:
                return False
            else:
                seens.add(n)
        return True