题目:
写一个算法来判断一个数是不是“快乐数”。
一个数是不是快乐是这么定义的:对于一个正整数,每一次将该数替换为它每个位置上的数字的平方和,然后重复这个过程直到这个数变为 1,或是无限循环但始终变不到 1。如果可以变为 1,那么这个数就是快乐数。
案例: 19 是一个快乐数。
- 12 + 92 = 82
- 82 + 22 = 68
- 62 + 82 = 100
- 12 + 02 + 02 = 1
分析:
- 分离n的每一位,计算平方和,若为1,则返回真,若不为1则将平方和赋值给n,继续循环;
- 使用集合记录过程中的出现的平方和,若平方和第二次出现,说明陷入死循环,返回False
代码:
class Solution(object):
def isHappy(self, n):
"""
:type n: int
:rtype: bool
"""
setTemp = []
while True:
sums = 0
while n // 10:
sums += (n % 10) ** 2
n //= 10
sums += (n % 10) ** 2
if sums == 1:
return True
n = sums
# print sums
# print setTemp
if sums in setTemp:
return False
setTemp.append(sums)
思考: