0202-leetcode算法实现之快乐数-happy-number-python&golang实现

编写一个算法来判断一个数 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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/happy-number

方法：

• 方法一：哈希表法-可以收敛到1，肯定是有限迭代，否则一定进入循环
• 方法二：快慢指针-类似环形链表快慢指针相遇问题

python

# 0202.快乐数
class Solution:
    def isHappy(self, n: int) -> bool:
        """
        哈希法
        思路：
        - 加进哈希表
        - 分离个位数，重新得新结果
        - n不为1或n不在哈希表中，重复上2步，n为1或n已在哈希表中(陷入循环重复)
        :param n:
        :return:
        """
        # 分离个位数，返回下一个数
        def get_next(n):
            total_sum = 0
            while n > 0:
                # 分离得到整除10及最小位数字 (x//y, x%y)
                n, digit = divmod(n, 10)
                total_sum += digit ** 2
            return total_sum

        seen = set()
        while n != 1 and n not in seen:
            seen.add(n)
            n = get_next(n)

        return n == 1

golang

func isHappy(n int) bool {
	m := map[int]bool{}
	for n != 1 && !m[n] {
		m[n] = true
		n = step(n)
	}
	return n == 1
}

func step(n int) int {
	sum := 0

	for n > 0 {
		sum += (n % 10) * (n % 10)
		n = n / 10
	}
	return sum
}