202. 快乐数

最新推荐文章于 2024-10-17 20:57:22 发布

百里守约学编程

最新推荐文章于 2024-10-17 20:57:22 发布

阅读量294

点赞数 5

分类专栏： # 哈希表文章标签： golang

本文链接：https://blog.csdn.net/YouMing_Li/article/details/142306829

版权

哈希表专栏收录该内容

8 篇文章 0 订阅

订阅专栏

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^31 - 1$

思路： 这题的关键其实就是在于什么时候可以退出循环，如果某一步计算出的结果发现曾经出现过了，说明后续结果也会一直不断重复，会进入了无限循环，此时我们应该退出了。而判断一个数字是否出现过，第一反应就是使用Set。因此思路就很清晰了，首先定义一个Set集合，用来存放计算后的平方和sum，如果sum在Set中已存在，即进入了死循环，则退出；如果sum不存在于Set，则将sum放入Set；直至找到平方和为1或者进入死循环就返回false。

Go代码

func isHappy(n int) bool {
    /*首先定义一个Set集合，用来存放计算后的平方和sum，如果sum在Set中已存在，即进入了死循环，则退出；
    如果sum不存在Set，则将sum放入Set；
    直至找到平方和为1或者进入死循环就返回false。*/

    set := make(map[int]bool)
    for n != 1 && !set[n] {
        set[n] = true // 旧n放入Set，表示出现过了
        n = getSum(n)  
    }
    return n == 1
}

func getSum(n int) int {
    sum := 0
    for n != 0 {
        num := n % 10 // 当前个位数上的数字
        sum += num * num
        n /= 10
    }
    return sum
}