leetcode-202. 快乐数

题目

编写一个算法来判断一个数 n 是不是快乐数。

「快乐数」定义为：对于一个正整数，每一次将该数替换为它每个位置上的数字的平方和，然后重复这个过程直到这个数变为 1，也可能是 无限循环 但始终变不到 1。如果 可以变为 1，那么这个数就是快乐数。

如果 n 是快乐数就返回 True ；不是，则返回 False 。

示例：

输入：19
输出：true
解释：
1^2 + 9^2 = 82
8^2 + 2^2 = 68
6^2 + 8^2 = 100
1^2 + 0^2 + 0^2 = 1

解题思路

好久不做题了真的脑子不转了，简单题目也毫无思路。。。。看了题解才直到怎么做。
首先寻找的过程中应该有3种情况：

1. 收敛为1
2. 不收敛到1，循环
3. 不收敛到1，发散（越来越大）

前2种情况代码比较好写，第3种情况不好判断，具体分析一下：

输入的数字位数	当前位数对应的最大数字	最大数字对应的下一个数字
1	9	81
2	99	162
3	999	243
…	…	…
12	12个9	972
13	13个9	1053(再下次会跌落到4位数行列)

可以看到大位数数字的下一个数字位数跌落是非常明显的，13位的数字下一个最大也是4位数，所以不会出现3的情况即发散越来越大的
排除情况3后代码就很好写了。
这里是用set作为字典检查循环，类似的循环检查还可以用快慢指针的方式（Eg: 环形链表）

成绩

时间>49.37%
空间>9.09%

代码

class Solution:
    def isHappy(self, n: int) -> bool:
        n_list = [int(item) for item in str(n)]
        check_set = set()
        cur_sum = sum([item ** 2 for item in n_list])
        while cur_sum != 1 and cur_sum not in check_set:
            check_set.add(cur_sum)
            n_list = [int(item) for item in str(cur_sum)]
            cur_sum = sum([item ** 2 for item in n_list])
        if cur_sum == 1:
            return True
        return False