python暴力算法快乐数

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

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

示例: 

输入: 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或4往上的倍数时会进入无限死循环，所以我加了一个count，在循环50次还没有结果时直接break

class Solution:
def isHappy(self, n: int) -> bool:
n_str = str(n)
count = 0
while True:
count += 1
s = 0
for i in n_str:
s =s + int(i)**2
if s != 1:
n_str = s
n_str = str(n_str)
elif s == 1:
return True
break
if count == 50:
break