编写一个算法来判断一个数 n
是不是快乐数。
「快乐数」 定义为:
- 对于一个正整数,每一次将该数替换为它每个位置上的数字的平方和。
- 然后重复这个过程直到这个数变为 1,也可能是 无限循环 但始终变不到 1。
- 如果这个过程 结果为 1,那么这个数就是快乐数。
如果 n
是 快乐数 就返回 true
;不是,则返回 false
。
示例 1:
输入:n = 19 输出:true 解释: 12 + 92 = 82 82 + 22 = 68 62 + 82 = 100 12 + 02 + 02 = 1
示例 2:
输入:n = 2 输出:false
提示:
1 <= n <= 231 - 1
哈希表解法
function isHappy(n: number): boolean {
const hashSet = new Set();
while (n !== 1 && !hashSet.has(n)) {
hashSet.add(n);
/** 将数值分割转为数组,123 => [1,2,3] */
const nums = [...`${n}`].map(Number);
/** 将每个元素的平方和相加, cur ** 2 表示平方 */
n = nums.reduce((prev, cur) => prev + cur ** 2, 0);
}
return n === 1;
}
快慢指针
/** 快慢指针 */
function isHappy2(n: number): boolean {
//计算平方和
function getNext(v: number) {
return [...`${v}`]
.map((n) => Number(n) ** 2)
.reduce((prev, cur) => prev + cur, 0);
}
// 慢指针,一次走一步
let slow = n;
// 快指针,一次走二步
let fast = getNext(n);
// 如果 fast不等于1,则最终一定会走入循环,则fast的值最后会=slow,证明不快乐
while (fast !== 1 && slow !== fast) {
//慢指针走一步,计算slow的平方和
slow = getNext(slow);
//快指针走二步,调用再次计算平方和。
fast = getNext(getNext(fast));
}
return fast === 1;
}