leetcode 探索 哈希表 快乐数

题目

编写一个算法来判断一个数 n 是不是快乐数。
「快乐数」定义为：对于一个正整数，每一次将该数替换为它每个位置上的数字的平方和，然后重复这个过程直到这个数变为 1，也可能是 无限循环 但始终变不到 1。如果 可以变为 1，那么这个数就是快乐数。
如果 n 是快乐数就返回 True ；不是，则返回 False 。
示例：

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

分析

这题一看应该用递归，因为每次操作的步骤都是一样的。

1. 计算入参数字的每个位上数字的平方和
2. 判断平方和是否为1
3. 若是1 结束递归返回true
4. 若不是1，继续递归。

这里要注意的是，给出的值是有可能造成无线循环的，怎样会循环呢？即如果一个数已经判断过了，但是又遇到了说明无线循环了，此时需要退出递归。

解法

// 记录循环，如果循环了，就不可能为快乐数了
var set = make(map[int]struct{})

func isHappy(n int) bool {
    var sum int
    for {
        sum += (n % 10) * (n % 10)
        n = n / 10
        if n == 0 {
            break
        }
    }
    if sum == 1 {
        return true
    }
    if _, ok := set[sum]; ok {
        return false
    }
    set[sum] = struct{}{}
    
    return isHappy(sum)
}