原题
写一个算法来判断一个数是不是"快乐数"。
一个数是不是快乐是这么定义的：对于一个正整数，每一次将该数替换为他每个位置上的数字的平方和，然后重复这个过程直到这个数变为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
