力扣刷题 202. 快乐数

目录

题干

数据结构

解题思路

---

题干

编写一个算法来判断一个数 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

---

解题思路

本题的核心点在于题干中提到如果这个数不是快乐数，那么就会陷入无限循环，这暗示了平方和sum必然会出现重复，因此我们可以以sum是否重复来作为不是快乐数的依据。因此我们想到了哈希表，因为哈希表最重要的作用就是快速判断一个元素是否出现集合里。 由于sum的数量是不固定的，采用数组会造成空间的大量浪费，所以我们这里采用unordered_set作为哈希表的数据结构。

因此，我们的思路是计算数 n 各位上的平方和，并判断是否为1，如果成立则 return true。如果不成立，判断该平方和是否在哈希表中，如果存在则说明陷入无限循环，return false，如果还没有重复，则将平方和放入哈希表，并更新数 n 的值继续计算平方和，进行到下一次循环。

笔者在一开始设置循环时，使用的判断条件是

while(!(result_set.find(n) != result_set.end())){

 一旦在set中找到相同元素就结束循环，但是这样的设计看似合理，但有一个致命的问题，即将平方和放入set和判断平方和是否在set中是有先后顺序的，如果先放入后判断，那么就会在set中找到自己，因此只能够在判断后再放入。现在的情况是放入在判断之前，所以一定会错误。我们设置while（1），在循环内进行判断和放入操作，就可以确定顺序了。

计算平方和是一个输入到结果的过程，可以用函数来简化。

本题的另一个难点在于如何提取数 n  的每一位并计算平方和，具体思路会在下面的代码详解中仔细说明。

完整代码如下：

class Solution {
public:
    int getsum(int n){
        int sum = 0;
        while(n){
            sum += (n % 10)*(n % 10);
            n /= 10;
        }
        return sum;
    }
    bool isHappy(int n) {
       unordered_set<int> set;
       while(1){
          int sum = getsum(n);
          if(sum == 1){
           return true;
           } 
          if(set.find(sum)!=set.end())
            return false;
          else
            set.insert(sum);
            n = sum;
       }
       
    }
};

---

 int getSum(int n) {
        int sum = 0;
        while (n) {
            sum += (n % 10) * (n % 10);
            n /= 10;
        }
        return sum;
    }

这个函数实现了计算数 n 每位平方和的功能。

n % 10 表示取整数 n 除以 10 后的余数，在这里n % 10 用于获取整数 n 的个位数字。

n /= 10的作用是将变量n整除以10，并将结果重新赋值给n，实现了删除最后一位的功能，例如223/10=22，每一次循环中，相当于往高位移动一位，直到只剩最高位，这时n/10就等于0，结束循环。这样配合上n%10这个表达式，我们就可以提取出数 n 的每一位啦。

但这里还有一个需要注意的地方，while的结束条件是遍历完数的每一位，那么该如何设置呢？

当n为一位数时，n/10的答案为0。（不是n，笔者经常与n%10混在一起）我们可以利用这一特点，设置while（n），这样当n为个位数时就会终止循环。