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 &lt;= n &lt;= 2<sup>31</sup><span> </span>- 1

Solution


第一反应,这不就是模运算吗。果断写下:

python 复制代码
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

分析一下,这是对于无限循环的情况缺少考虑。

无限循环的情况是什么呢?就是检测之前是否出现过。毕竟是循环。所以如果之前出现过的数集。

python 复制代码
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)

但是:

python 复制代码
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)
python 复制代码
✓ 2 tests complete
× @test(7)  result: false ,expect: true

这次的问题出在哪里呢?

原代码中的 while n >= 10 条件会导致个位数被忽略。例如,对于数字 19,代码会计算 9^2 = 81,但会忽略 1^2。

python 复制代码
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, 通过。

相关推荐
ting_zh15 小时前
导数、偏导数与梯度:机器学习数学基础
算法·基础数学
灰灰老师15 小时前
七种排序算法比较与选择[Python ]
java·算法·排序算法
杰瑞学AI15 小时前
我的全栈学习之旅:FastAPI (持续更新!!!)
后端·python·websocket·学习·http·restful·fastapi
用户37215742613515 小时前
Python 高效实现 Excel 与 CSV 互转:用自动化提升效率
python
秃头狂魔15 小时前
DAY1 数组一
算法
CM莫问15 小时前
推荐算法之粗排
深度学习·算法·机器学习·数据挖掘·排序算法·推荐算法·粗排
rengang6616 小时前
10-支持向量机(SVM):讲解基于最大间隔原则的分类算法
人工智能·算法·机器学习·支持向量机
CodeCraft Studio16 小时前
CAD文件处理控件Aspose.CAD教程:在 Python 中将 SVG 转换为 PDF
开发语言·python·pdf·svg·cad·aspose·aspose.cad
mortimer16 小时前
从预处理到合成:基于pySide6的视频翻译多线程流水线架构详解
python·github
喜欢吃豆16 小时前
从潜在空间到实际应用:Embedding模型架构与训练范式的综合解析
python·自然语言处理·架构·大模型·微调·embedding