编写一个算法来判断一个数是不是“快乐数”。
一个“快乐数”定义为:对于一个正整数,每一次将该数替换为它每个位置上的数字的平方和,然后重复这个过程直到这个数变为 1,也可能是无限循环但始终变不到 1。如果可以变为 1,那么这个数就是快乐数。
示例:
输入: 19
输出: true
解释:
12 + 92 = 82
82 + 22 = 68
62 + 82 = 100
12 + 02 + 02 = 1
思路:
照着要求算就行了,开辟一个list来储存之前出现过的值,如果当前值之前已经出现过,就说明陷入循环,肯定不可能变成1。
class Solution(object):
def isHappy(self, n):
"""
:type n: int
:rtype: bool
"""
cnt = 0
l = []
while(n!= 1):
l.append(n)
temp = 0
while(n > 0):
temp += (n % 10) ** 2
n /= 10
n = temp
if n in l:
return False
return n == 1
下面的写于2019年7月24日13:27:05
class Solution(object):
def isHappy(self, n):
"""
:type n: int
:rtype: bool
"""
def happy(num):
res = 0
while num:
num, tmp = divmod(num, 10)
res += tmp ** 2
return res
visited = set()
while n and n not in visited:
visited.add(n)
tmp = happy(n)
if tmp == 1:
return True
n = tmp
return False