【leetcode 簡單】 第五十六題 快樂數

編寫一個演算法來判斷一個數是不是“快樂數”。

一個“快樂數”定義為：對於一個正整數，每一次將該數替換為它每個位置上的數字的平方和，然後重覆這個過程直到這個數變為 1，也可能是無限迴圈但始終變不到 1。如果可以變為 1，那麼這個數就是快樂數。

class Solution:
    def isHappy(self, n):
        """
        :type n: int
        :rtype: bool
        """
        happySet = set([1, 7, 10, 13, 19, 23, 28, 31, 32, 44, 49, 68, 70, 79, 82, 86, 91, 94, 97])
        while n>99:
            n = sum([int(x) * int(x) for x in list(str(n))])
        return n in happySet