Leetcode 202

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

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

示例: 

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

方法:

根据题意,需要重点关注的是计算过程中会不会陷入无限循环,如果可以等于1 直接返回True。因此最好的选择就是hash表,python中的字典或者直接使用集合。

注意:本次代码使用lambda 表达式,这也是python的一个重点内容,最近在复习装饰器,过段时间会写一个Blog分析相关内容

最终的AC代码为

class Solution:
    def isHappy(self, n):
        """
        :type n: int
        :rtype: bool
        """
        data = list(map(int, str(n)))
        num = 0
        s = set()
        while num != 1:
            num = sum(list(map(lambda x: x**2, data)))
            if num not in s:
                s.add(num)
            else:
                return False
            data = list(map(int, str(num)))
        return True

猜你喜欢

转载自blog.csdn.net/jhlovetll/article/details/85645388
今日推荐