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

「快乐数」 定义为:
对于一个正整数,每一次将该数替换为它每个位置上的数字的平方和。
然后重复这个过程直到这个数变为 1,也可能是 无限循环 但始终变不到 1。
如果这个过程 结果为 1,那么这个数就是快乐数。
如果 n 是 快乐数 就返回 true ;不是,则返回 false 。
示例: :::info 输入:n = 19
输出:true
解释:
12 + 92 = 82
82 + 22 = 68
62 + 82 = 100
12 + 02 + 02 = 1 ::: 来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/happy-number
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

思路

本题值得注意的点在于,出现无限循环的情况。为了解决这个问题,我们可以使用哈希表set去记录每一个数字的平方和,当出现重复数字时,直接返回false;当出现结果为 1 时,则返回 true.

  1. var getSum = function (n) {
  2. let sum = 0;
  3. while (n) {
  4. sum += (n % 10) ** 2;
  5. n = Math.floor(n/10);
  6. }
  7. return sum;
  8. }
  9. var isHappy = function(n) {
  10. let set = new Set(); // Set() 里的数是惟一的
  11. // 如果在循环中某个值重复出现,说明此时陷入死循环,也就说明这个值不是快乐数
  12. while (n !== 1 && !set.has(n)) {
  13. set.add(n);
  14. n = getSum(n);
  15. }
  16. return n === 1;
  17. };