leetcode_414 第三大的数

83 阅读1分钟

要求

给你一个非空数组,返回此数组中 第三大的数 。如果不存在,则返回数组中最大的数。

示例 1:

输入:[3, 2, 1]
输出:1
解释:第三大的数是 1

示例 2:

输入:[1, 2]
输出:2
解释:第三大的数不存在, 所以返回最大的数 2

示例 3:

输入:[2, 2, 3, 1]
输出:1
解释:注意,要求返回第三大的数,是指在所有不同数字中排第三大的数。
此例中存在两个值为 2 的数,它们都排第二。在所有不同数字中排第三大的数为 1

核心代码

class Solution(object):
    def thirdMax(self, nums):
        A = sorted(list(set(nums)))
        if len(A) < 3:
            return max(A)
        else:
            A.reverse()
            return A[2]

另一解法

class Solution(object):
    def thirdMax(self, nums):
        res = [float("-inf")] * 3
        for i in nums:
            if i in res:
                continue
            if i > res[0]:
                res = [i,res[0],res[1]]
            elif i > res[1]:
                res = [res[0],i,res[1]]
            elif i > res[2]:
                res = [res[0],res[1],i]
        return res[-1] if res[2] != float("-inf") else res[0]

image.png

解题思路:第一种解法:使用排序的方式完成第三大数字的查找;第二种解法:我们使用一个列表只保留最大的三个数字,然后,循环遍历,不断更新,最终获取三个数字中最小的那个,比较简单。