LeetCode 202 [Happy Number]

原题

写一个算法来判断一个数是不是"快乐数"。
一个数是不是快乐是这么定义的：对于一个正整数，每一次将该数替换为他每个位置上的数字的平方和，然后重复这个过程直到这个数变为1，或是无限循环但始终变不到1。如果可以变为1，那么这个数就是快乐数。

样例
19 就是一个快乐数。

1^2 + 9^2 = 82
8^2 + 2^2 = 68
6^2 + 8^2 = 100
1^2 + 0^2 + 0^2 = 1

解题思路

• 首先写一个helper函数，计算一个数每一位的平方和。比如输入19返回82
• 当n不等于1且n计算过后的值未出现过，继续计算。若最后出现循环则返回False

完整代码

class Solution(object):
    def isHappy(self, n):
        """
        :type n: int
        :rtype: bool
        """
        if n < 1:
            return False
        myset = set()
        while n != 1 and n not in myset:
            myset.add(n)
            n = self.helper(n)
        return n == 1
        
    def helper(self, num):
        res = 0
        while num:
            res += (num % 10) * (num % 10)
            num /= 10
        return res