[LeetCode] 202. 快乐数
题目链接: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
思路:
这道题难点,在于如何解决循环重复的情况?
备注:我这里都是用字符串求和,也可以用数学方法(取余那种)
方法一:用哈希记录
class Solution:
def isHappy(self, n: int) -> bool:
n = str(n)
visited = set()
while 1:
n = str(sum(int(i) ** 2 for i in n))
if n == "1":
return True
if n in visited:
return False
visited.add(n)
方法二:快慢(跑)
class Solution:
def isHappy(self, n: int) -> bool:
n = str(n)
slow = n
fast = str(sum(int(i) ** 2 for i in n))
while slow != fast:
slow = str(sum(int(i) ** 2 for i in slow))
fast = str(sum(int(i) ** 2 for i in fast))
fast = str(sum(int(i) ** 2 for i in fast))
return slow == "1"