Leetcode - 202 快乐数

 

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

「快乐数」 定义为：

对于一个正整数，每一次将该数替换为它每个位置上的数字的平方和。

然后重复这个过程直到这个数变为 1，也可能是 无限循环 但始终变不到 1。

如果这个过程 结果为 1，那么这个数就是快乐数。

如果 n 是 快乐数 就返回 true ；不是，则返回 false 。

示例 ：

输入：n = 19
输出：true
解释：
1^2 + 9^2 = 82
8^2 + 2^2 = 68
6^2 + 8^2 = 100
1^2 + 0^2 + 0^2 = 1

class Solution:
    def isHappy(self, n: int) -> bool:
        if n == 1:
            return True
        visited = set()  # 用于记录已经访问过的数字
        while n != 1 and n not in visited:
            visited.add(n)
            n = sum(int(digit) ** 2 for digit in str(n))
        return n == 1

• visited = set()：这行代码创建了一个空的集合visited。集合是Python中的一种数据结构，它允许我们存储不重复的元素。
• while n != 1 and n not in visited:：这是一个循环条件，它确保只有当n不等于1且n还未被访问时，循环才会继续执行。
• visited.add(n)：这行代码将当前数字n添加到visited集合中，以标记它已经被访问过。
• n = sum(int(digit) ** 2 for digit in str(n))：这行代码计算n的每个数字的平方和，并将结果重新赋给变量n。这里使用了Python的生成器表达式（for digit in str(n)）来遍历n的每个数字。int(digit) ** 2计算每个数字的平方，sum()函数则将这些平方值相加得到新的数字。
• return n == 1：最后，根据n是否等于1来判断是否为快乐数，并返回结果。如果n等于1，那么返回True（表示是快乐数），否则返回False。