leetcode_202. 快乐数python

题目描述

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

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

示例:

输入: 19
输出: true
解释:
1² + 9² = 82
8² + 2² = 68
6² + 8² = 100
1² + 0² + 0² = 1

算法思想

1.如果是快乐数的话,那么拆分加和到最后的话,变为1,程序结束
2.如果是不快乐的数,他会进行一个循环,这点要清楚,那么用什么进行判断呢,这时就要使用一个字典来存储每次数字做的加和,并且每次循环判断是否在列表中

代码1

class Solution(object):
    def isHappy(self, n):
        """
        :type n: int
        :rtype: bool
        """
        d = {}
        while True:
            l = list(map(int,list(str(n))))
            m = 0
            for i in l:
                m += i**2
            if m in d:
                print(d)
                return False
            if m == 1:
                print(d)
                return True
            d[m] = m
            n = m

代码2

class Solution(object):
    def isHappy(self, n):
        """
        :type n: int
        :rtype: bool
        """
        d = {}
        while True:
            m = 0
            while n:
                m += (n%10)**2
                n /= 10
            if m == 1:
                return True
            if m in d:
                return False
            d[m] = m
            n = m

代码3(使用列表存储以后效率大大的提升)

class Solution(object):
    def isHappy(self, n):
        """
        :type n: int
        :rtype: bool
        """
        d = []
        while True:
            m = 0
            while n:
                m += (n%10)**2
                n /= 10
            if m == 1:
                return True
            if m in d:
                return False
            d.append(m)
            n = m

知识点

1..map()函数的使用:
描述
map() 会根据提供的函数对指定序列做映射。
第一个参数 function 以参数序列中的每一个元素调用 function 函数,返回包含每次 function 函数返回值的新列表。
语法

map(function, iterable, ...)

参数

function – 函数
iterable – 一个或多个序列

实例

>>>def square(x) : # 计算平方数
... return x ** 2
...
>>> map(square, [1,2,3,4,5]) # 计算列表各个元素的平方
[1, 4, 9, 16, 25]
>>> map(lambda x: x ** 2, [1, 2, 3, 4, 5]) # 使用 lambda 匿名函数
[1, 4, 9, 16, 25]
# 提供了两个列表,对相同位置的列表数据进行相加
>>> map(lambda x, y: x + y, [1, 3, 5, 7, 9], [2, 4, 6, 8, 10])
[3, 7, 11, 15, 19]

猜你喜欢

转载自blog.csdn.net/qq_37002901/article/details/88682086