题目描述

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

「快乐数」定义为：对于一个正整数，每一次将该数替换为它每个位置上的数字的平方和，然后重复这个过程直到这个数变为 1，也可能是 无限循环 但始终变不到 1。如果 可以变为 1，那么这个数就是快乐数。

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

提示：

1 <= n <= 2^31 - 1

测试用例

• 输入：19
• 输出：true
• 解释：

1^2 + 9^2 = 82

8^2 + 2^2 = 68

6^2 + 8^2 = 100

1^2 + 0^2 + 0^2 = 1

代码实现

/**
 * @param {number} n
 * @return {boolean}
 */
var isHappy = function (n) {
  if (n === 1) {
    return true;
  }
  let set = new Set();
  while (n !== 1 && !set.has(n)) {
    set.add(n);
    n = getN(n);
  }
  return n === 1;
};

let getN = (n) => {
  if (n === 1 || n === 0) {
    return n;
  }
  let res = 0;
  while (n) {
    res += (n % 10) * (n % 10);
    n = parseInt(n / 10);
  }
  return res;
};

var __readline = require("readline-sync");
__readline.setDefaultOptions({ prompt: "" });
var readline = __readline.prompt;

console.log("请输入n: ");
const n = parseInt(readline());

const res = isHappy(n);
res && console.log(`${n}是快乐数`);
!res && console.log(`${n}不是快乐数`);

/* 
  请输入n:
  19
  19是快乐数

  请输入n:
  2
  2不是快乐数
*/