题目
编写一个算法来判断一个数 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,循环
- 不收敛到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