编写一个算法来判断一个数 n 是不是快乐数。
「快乐数」定义为:
对于一个正整数,每一次将该数替换为它每个位置上的数字的平方和。
然后重复这个过程直到这个数变为 1,也可能是 无限循环 但始终变不到 1。
如果 可以变为 1,那么这个数就是快乐数。
如果 n 是快乐数就返回 true ;不是,则返回 false 。
示例 1:
输入:19
输出:true
解释:
12 + 92 = 82
82 + 22 = 68
62 + 82 = 100
12 + 02 + 02 = 1
示例 2:
输入:n = 2
输出:false
提示:
1 <= n <= 231 - 1
这道题目看上去貌似一道数学问题,其实并不是!
题目中说了会 无限循环,那么也就是说求和的过程中,sum会重复出现,这对解题很重要!
正如:关于哈希表,你该了解这些!中所说,当我们遇到了要快速判断一个元素是否出现集合里的时候,就要考虑哈希法了。
所以这道题目使用哈希法,来判断这个sum是否重复出现,如果重复了就是return false, 否则一直找到sum为1为止。
方法一:用哈希集合检测循环
算法分为两个部分
- 要求数位分离求平方和
- 使用哈希集合完成,每次生产链中的下一个数字时,我们都会检查它是否已经在哈希集合中
- 如果不在哈希集合中,我们应该添加它
- 如果在哈希集合中,意味着我们处于一个循环之中,因此返回false
代码如下:
/**
* @param {number} n
* @return {boolean}
*/
function getN(n) {
let res = 0;
while (n) {
res += (n % 10) * (n % 10);
n = Math.floor(n / 10);
}
return res;
}
var isHappy = function(n) {
const sumSet = new Set();
while (n != 1) {
if(sumSet.has(n)){
return false;
}
sumSet.add(n);
n = getN(n);
}
return true;
};
补充:集合中一些可用的方法
- add(element):向集合添加一个新元素
- has(element):如果元素在集合中,返回true,否则返回false
- delete(element):从集合移除一个元素
- clear()
- size():与数组的length属性相似
- values():返回一个包含集合中所有值的数组