编写算法以确定数字n是否为“ happy”。
一个快乐的数字是由以下过程定义的数字:从任何正整数开始,用该数字的平方和代替该数字,然后重复该过程,直到该数字等于1(它将停留在该位置),或它以不包含1的循环无限循环。以1结尾的那些数字是快乐数字。
如果n是一个快乐数字,则返回True;否则返回False。
Input: 19
Output: True(1²+9²=82 8²+2²=62 6²+2²=100 1²+0²+0²=1)
Input: 2
Output:False(2²=4)
Clue:n等于n的每个位的平方加和。定义一个临时list(temp)用来存储n的值,如果n存在temp里面就跳出循环。
class Solution:
def isHappy(self, n: int) -> bool:
temp=[]
while n!=1 and n not in temp:
temp.append(n)
n=sum(int(i)**2 for i in str(n))
#s=0
#for i in str(n):
# s+=int(i)**2
#n=s
return n==1
'''
4 [2]
16 [2, 4]
37 [2, 4, 16]
58 [2, 4, 16, 37]
89 [2, 4, 16, 37, 58]
145 [2, 4, 16, 37, 58, 89]
42 [2, 4, 16, 37, 58, 89, 145]
20 [2, 4, 16, 37, 58, 89, 145, 42]
4 [2, 4, 16, 37, 58, 89, 145, 42, 20]
False
'''
注释里是以n=2为例,输出的结果和sum( )的替换