23-代码随想录第202题快乐数

202. 快乐数

简单
相关标签
相关企业
编写一个算法来判断一个数 n 是不是快乐数。

「快乐数」定义为：

对于一个正整数，每一次将该数替换为它每个位置上的数字的平方和。
然后重复这个过程直到这个数变为 1，也可能是无限循环但始终变不到 1。
如果这个过程结果为 1，那么这个数就是快乐数。
如果 n 是快乐数就返回 true ；不是，则返回 false 。

示例 1：

输入：n = 19
输出：true
解释：
12 + 92 = 82
82 + 22 = 68
62 + 82 = 100
12 + 02 + 02 = 1
示例 2：

输入：n = 2
输出：false

提示：

1 <= n <= 231 - 1

代码

def ishappy(n):
    record=set()
    while True:
        n=getsum(n)
        if n==1:
            print(True)
            return
        if n in record:
            print(False)
            return
        else:
            record.add(n)

def getsum(n):
    new_num=0
    while n:
        n,r=divmod(n,10)#将n除以10得到商和余数，商赋值给变量n，余数赋值给变量r
        new_num+=r**2
    return new_num

def myhappy(n):
    sum=0
    record=set()
    for i in range(len(n)):
        sum+=n[i]**n[i]
        if sum==1:
            print(True)
            return
        if sum in record or i==len(n)-1:
            print(False)
            return
        else:
            record.add(sum)

if __name__=='__main__':
    n=input("n=").split()
    n= [int(x) for x in n if x.isdigit()]
    myhappy(n)#用了列表
    n=int(input("n="))
    ishappy(n)#用了divmod这个神奇的函数