编写一个算法来判断一个数 n 是不是快乐数。
「快乐数」定义为:
对于一个正整数,每一次将该数替换为它每个位置上的数字的平方和。
然后重复这个过程直到这个数变为 1,也可能是 无限循环 但始终变不到 1。
如果 可以变为 1,那么这个数就是快乐数。
如果 n 是快乐数就返回 true ;不是,则返回 false 。
示例 1:
输入:19
输出:true
解释:
1 * 1 + 9 * 9 = 82
8 * 8 + 2 * 2 = 68
6 * 6 + 8 * 8 = 100
1 * 1 + 0 * 0 + 0 * 0 = 1
示例 2:
输入:n = 2
输出:false
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/happy-number
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
# 初始化res为0用来保存每次各个位数上的数字平方和,初始化列表x用来保存计算过程中非快乐数的数字,建立无限循环,首先计算各个位数上的数字平方和,若其 == 1则为快乐数,若其 in x则不是,若以上条件都不满足则给n赋新值以便进行下次运算,将n添加至x以便进行下次对比,使res为0以便保存下次循环各个位数上的数字平方和
class Solution:
def isHappy(self, n: int) -> bool:
res = 0
x = [n] # 保存非快乐数
while True:
for i in str(n):
res += int(i)**2 # 求出各个位数上的数字平方和
if res == 1:
return True
elif res in x: # 即res为非快乐数
return False
else:
n = res # 若直接将res添加至x会导致变量一直变化,所以应该先将res值赋予n,再添加至x
x.append(n)
res = 0