「leetcode」202 - 快乐数

145 阅读1分钟

链表相关第三篇,前两篇见:

  1. 「leetcode」141-环形链表(两种解法&JS实现的注意点)
  2. 「leetcode」206-反转链表

题目

编写一个算法来判断一个数 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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/happy-number
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

解法

  • 链表应用题~可以理解为n是val, 组合出来的新val是next的value
  • 无限循环 == 环形链表存在;变为1 == 终止条件
  • 环形链表 -> 快慢指针
  • 代码出现
/**
 * @param {number} n
 * @return {boolean}
 */
var isHappy = function(n) {
  // 获取next
  function getNext(head) {
    if (head === 0) {return 0}
    let result = 0
    head.toString().split('').map(i => i*i).map(i => result += i)
    return result
  }
  
  if (n === 0) { return false }
  let slow = n // 慢指针
  let quick = getNext(getNext(n)) // 快指针
  if (slow === 1 || quick === 1) return true
  while(slow !== quick && quick !== 1) {
    slow = getNext(slow)
    quick = getNext(getNext(quick))
  }
  if (slow === 1 || quick === 1) return true
  return false
};