经典算法题（二）—快乐数-优快云博客

本文链接：https://blog.youkuaiyun.com/weixin_48232453/article/details/142460860

经典算法题（二）—快乐数

问题描述

判断一个数是否为快乐数。

快乐数 定义为：

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

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

示例

示例	输入	输出	解释
示例1	19	True	1x1 + 9x9 =82 8x8 + 2x2=68 6x6 + 8x8=100 1x1+0x0+0x0=1
示例2	2	False

算法思路

1.快乐数的判定条件：

如果某个数字最终变为 1，那么它是快乐数；
如果出现循环，即某个数字已经出现过，但不是 1，那么它就不是快乐数。

2.如何检测循环：

可以使用 哈希集合（set） 来存储每次出现的平方和，若平方和再次出现，说明进入了循环，此时可以返回 False。

关键步骤：

初始化一个哈希集合，用来记录已经出现过的数字。
对给定的数字 n，不断计算其各位数字的平方和。
如果平方和为 1，则返回 True，表示这是一个快乐数。
如果平方和已经在哈希集合中出现过，返回 False，表示进入了无限循环，n 不是快乐数。

代码实现

# 我的代码：
def isHappy(n: int) -> bool:
    def get_next_number(num):
        total_sum = 0
        while num > 0:
            digit = num % 10
            total_sum += digit ** 2
            num //= 10
        return total_sum
    
    seen = set()
    while n != 1 and n not in seen:
        seen.add(n)
        n = get_next_number(n)
    
    return n == 1

# 官方代码：
def isHappy(n: int) -> bool:
    def get_next(n):
        while(n>0):
            n,digit = divmod(n,10)
            count += digit **2
        return count

    seen = set()
    while n!=1 and n not in seen:
        seen.append(n)
        n = get_next(n)

    return n == 1