快乐数——使用哈希集合求解

1题目

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

「快乐数」 定义为：

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

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

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

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

示例 1：

输入：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

示例 2：

输入：n = 2

输出：false

提示：

• 1 <= n <= 231 - 1

2链接

题目：202. 快乐数 - 力扣（Leetcode）

3解题思路

当我们遇到了要快速判断一个元素是否出现集合里的时候，就要考虑哈希法了。

所以这道题目使用哈希法，来判断这个sum是否重复出现，如果重复了就是return false， 否则一直找到sum为1为止。

判断sum是否重复出现就可以使用unordered_set。

注意一个数每一位的和如何去求，否则会造成解题困难

注意可能出现如下死循环情况，应及时return false

4代码

class Solution {
public:
    //获取每一位数的和
    int getSum (int n) {
        int sum = 0;
        while (n != 0) {
            sum += (n % 10) * (n % 10);
            n/=10;
        }
        return sum;
    }
    bool isHappy(int n) {
        unordered_set<int> set;
        int sum = 0;
        while (1) {
            sum = getSum(n);//每轮循环都会初始化sum
            if (sum == 1) return true;        
            //sum储存在set中的值每次都不一样，如果从某一次开始找到了曾出现过的sum
            //说明已经陷入了无限循环了，立刻return false
            if (set.find(sum) != set.end()) {
                return false;
            }
            //如果sum不为1，且未曾出现过set中，则继续往set中插入
            else {
                set.insert(sum);
            }
            n = sum;
        }
    }
};