【问题描述】
编写一个算法来判断一个数是不是“快乐数”。 一个“快乐数”定义为:对于一个正整数,每一次将该数替换为它每个位置上的数字的平方和,然后重复这个过程直 到这个数变为 1,也可能是无限循环但始终变不到 1。如果可以变为 1,那么这个数就是快乐数。
【样例输入】
32
【样例输出】
yes
【样例输入】
37
【样例输出】
no
def happy(n):
result=0
while(n):
result+=(n%10)**2
n=n//10
return result
s=set()
n=int(input())
while True:
n=happy(n)
if n==1:
print("yes")
break
elif n in s:
print("no")
break
else:
s.add(n)