LeetCode 202. 快乐数 (Happy Number)[简单]

lework · 2020年04月27日 · 最后由 lework 回复于 2020年04月27日 · 162 次阅读

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

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

如果 n 是快乐数就返回 True ;不是,则返回 False 。

 

示例:

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

来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/happy-number 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

class Solution:
    def isHappy(self, n: int) -> bool:
        mem = set()                #利用 set()集合,保存平方求和后的数值
        while n != 1:              #利用 while 实现循环      
            n = sum([int(i)**2 for i in str(n)])    #通过str(n)调取输入整数各个位数的值
            if n not in mem:       #若平方求和后的数值是首次出现,则添加进集合中
                mem.add(n)
            else:                  #若求和后数值,在集合中存在,则直接返回false,即出现死循环
                return False
        return True                #最终当n==1时,跳出while循环,返回true
需要 登录 后方可回复, 如果你还没有账号请点击这里 注册