题目来源:https://leetcode-cn.com/problems/happy-number/
编写一个算法来判断一个数是不是“快乐数”。
一个“快乐数”定义为:对于一个正整数,每一次将该数替换为它每个位置上的数字的平方和,然后重复这个过程直到这个数变为 1,也可能是无限循环但始终变不到 1。如果可以变为 1,那么这个数就是快乐数。
示例:
输入: 19
输出: true
解释:
12 + 92 = 82
82 + 22 = 68
62 + 82 = 100
12 + 02 + 02 = 1
解法一:双指针-快慢指针
根据题意,快乐数就是按照计算规则一直循环下去,最终能收敛到1,之后就一直循环1,而非快乐数就是无法收敛到1的数,也就是说会无限循环。
那么怎么跳脱出这个无限循环呢?
可以用快慢指针来做,慢指针每次走一步,快指针每次走两步,这样不断循环,一定会出现某一个时刻快指针等于慢指针,如果此时值为1,说明为快乐数,否则就不是快乐数。
class Solution(object):
def isHappy(self, n):
"""
:type n: int
:rtype: bool
"""
if n == 1:
return True
if n == 0:
return False
slow = n
fast = n
while slow != 1 and fast != 1:
slow = sum([int(i)**2 for i in str(slow)]) # 慢指针,走一步
fast = sum([int(i)**2 for i in str(fast)]) # 快指针,走两步
fast = sum([int(i)**2 for i in str(fast)])
if slow == fast and slow != 1 and fast != 1: # 两个指针遇上了,但是却不为1,说明不是快乐数
return False
return True
解法二:利用set的非重复性
非快乐数会陷入非1的循环,那么利用set元素的非重复性,出现第一个重复元素时就结束循环,判断此时和是否为1。
class Solution(object):
def isHappy(self, n):
"""
:type n: int
:rtype: bool
"""
s = {1}
while n not in s: # 出现第一个重复元素时,结束循环
s.add(n)
n = sum([int(i)**2 for i in str(n)])
return n == 1
参考:
https://leetcode-cn.com/problems/happy-number/solution/kuai-le-shu-kuai-man-zhi-zhen-by-wjx_seu2017/
https://leetcode-cn.com/problems/happy-number/solution/python-1xing-by-knifezhu-9/