Leetcode 202 快乐数
Leetcode 202 快乐数
编写一个算法来判断一个数 n
是不是快乐数。
「快乐数」 定义为:
- 对于一个正整数,每一次将该数替换为它每个位置上的数字的平方和。
- 然后重复这个过程直到这个数变为 1,也可能是 无限循环 但始终变不到 1。
- 如果这个过程 结果为 1,那么这个数就是快乐数。
如果 n
是 快乐数 就返回 true
;不是,则返回 false
。
示例 1:
输入:n = 19
输出:true
解释:
12 + 92 = 82
82 + 22 = 68
62 + 82 = 100
12 + 02 + 02 = 1
示例 2:
输入:n = 2
输出:false
提示:
1 <= n <= 2<sup>31</sup><span> </span>- 1
第一反应,这不就是模运算吗。果断写下:
class Solution:
def isHappy(self, n: int) -> bool:
if n == 1: return True
res = 0
while n >= 10:
res += (n%10)**2
n = n % 10
return self.isHappy(res)
然后果断报错。
RecursionError: maximum recursion depth exceeded
分析一下,这是对于无限循环的情况缺少考虑。
无限循环的情况是什么呢?就是检测之前是否出现过。毕竟是循环。所以如果之前出现过的数集。
class Solution:
def __init__(self) -> None:
self.visited = set()
def isHappy(self, n: int) -> bool:
if n == 1: return True
if n in self.visited:return False
self.visited.add(n)
res = 0
while n >= 10:
res += (n%10)**2
n = n % 10
return self.isHappy(res)
但是:
class Solution:
def __init__(self) -> None:
self.visited = set()
def isHappy(self, n: int) -> bool:
if n == 1: return True
if n in self.visited:return False
self.visited.add(n)
res = 0
while n >= 10:
res += (n%10)**2
n = n % 10
return self.isHappy(res)
✓ 2 tests complete
× @test(7) result: false ,expect: true
这次的问题出在哪里呢?
原代码中的 while n >= 10 条件会导致个位数被忽略。例如,对于数字 19,代码会计算 9^2 = 81,但会忽略 1^2。
class Solution:
def __init__(self) -> None:
self.visited = set()
def isHappy(self, n: int) -> bool:
if n == 1: return True
if n in self.visited:return False
self.visited.add(n)
res = 0
while n > 0:
res += (n%10)**2
n = n // 10
return self.isHappy(res)
ok, 通过。
原文地址:https://blog.csdn.net/qq_34271349/article/details/142920558
免责声明:本站文章内容转载自网络资源,如本站内容侵犯了原著者的合法权益,可联系本站删除。更多内容请关注自学内容网(zxcms.com)!