202.快乐数.go

/*
 * @lc app=leetcode.cn id=202 lang=golang
 *
 * [202] 快乐数
 *
 * https://leetcode-cn.com/problems/happy-number/description/
 *
 * algorithms
 * Easy (62.50%)
 * Likes:    900
 * Dislikes: 0
 * Total Accepted:    234.8K
 * Total Submissions: 374.8K
 * Testcase Example:  '19'
 *
 * 编写一个算法来判断一个数 n 是不是快乐数。
 *
 * 「快乐数」 定义为：
 *
 *
 * 对于一个正整数，每一次将该数替换为它每个位置上的数字的平方和。
 * 然后重复这个过程直到这个数变为 1，也可能是 无限循环 但始终变不到 1。
 * 如果这个过程 结果为 1，那么这个数就是快乐数。
 *
 *
 * 如果 n 是 快乐数 就返回 true ；不是，则返回 false 。
 *
 *
 *
 * 示例 1：
 *
 *
 * 输入：n = 19
 * 输出：true
 * 解释：
 * 1^2 + 9^2 = 82
 * 8^2 + 2^2 = 68
 * 6^2 + 8^2 = 100
 * 1^2 + 0^2 + 0^2 = 1
 *
 *
 * 示例 2：
 *
 *
 * 输入：n = 2
 * 输出：false
 *
 *
 *
 *
 * 提示：
 *
 *
 * 1 <= n <= 2^31 - 1
 *
 *
 */

// @lc code=start
func isHappy(n int) bool {
	// 为处理过的数字建立缓存, 防止发生死循环
	cache := make(map[int]bool)

	for n != 1 {
		if cache[n] {
			return false
		} else {
			cache[n] = true
			n = getNew(n)
		}
	}
	return true
}

func getNew(n int) int {
	ret := 0
	for n != 0 {
		last := n % 10
		ret += last * last
		n /= 10
	}
	return ret
}

// @lc code=end