leetcode 202. 快乐数（python）

题目链接

题目描述：
编写一个算法来判断一个数是不是“快乐数”。

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

示例:

输入: 19
输出: true
解释:
12 + 92 = 82
82 + 22 = 68
62 + 82 = 100
12 + 02 + 02 = 1

解题思路：
定义一个happynum函数用来计算每个数的各位平方之和m，若m不等于1，判断当前m是否出现过（无限循环），若出现过则说明该数不是快乐数，返回false，若没有出现过则继续调用该函数；若m等于1，说明是快乐数，返回true

def happynum(n):
    m=0
    for i in str(n):
        m=m+int(i)**2
    return m

class Solution:
    def isHappy(self, n: int) -> bool:
        res=[]
        m=happynum(n)
        while m!=1:
            if m in res:
                return False
            else:
                res.append(m)
            m=happynum(m)
        return True