【Leetcode202】快乐数

题目描述

编写一个算法来判断一个数 n 是不是快乐数。

「快乐数」 定义为：

对于一个正整数，每一次将该数替换为它每个位置上的数字的平方和。
然后重复这个过程直到这个数变为 1，也可能是 无限循环 但始终变不到 1。
如果这个过程 结果为 1，那么这个数就是快乐数。
如果 n 是 快乐数 就返回 true ；不是，则返回 false 。

思路与算法

1. 将“计算数字各位平方和”的逻辑封装到单独的函数中。
2. 可以通过将数字转换为字符串，然后逐位计算平方和
3. 问题理解：给定一个正整数 n，每次将 n 替换为其各位数字的平方和，判断最终是否会得到 1。如果过程无限循环而不出现 1，则 n 不是快乐数。
4. 检测循环问题：可以使用集合记录出现过的数字

代码

class Solution:

    def isHappy(self, n: int) -> bool:
        # 计算n的每位数字的平方和
        def next_num(n:int) -> int:
            return sum(int(digit)**2 for digit in str(n))
        
        # 用于记录已经出现的数字
        seen = set()

        while n not in seen and n != 1:
            seen.add(n)
            n = next_num(n)
        
        return n == 1

总结