202. 快乐数

202. 快乐数

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^{3}1-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 := 0for n != 0 {num := n % 10 // 当前个位数上的数字sum += num * numn /= 10}return sum
}