leetcode414.第三大的数

128 阅读1分钟

414.第三大的数

给定一个非空数组,返回此数组中第三大的数。若不存在,返回其中最大的数。时间复杂度O(n)。

输入: [2, 2, 3, 1]         输出: 1
解释: 注意,要求返回第三大的数,是指第三大且唯一出现的数。
存在两个值为2的数,它们都排第二。
class Solution:
    def thirdMax(self, nums: List[int]) -> int:
        set_nums = set(nums)
        if len(set_nums) < 3:
            return max(set_nums)
        else:
            set_nums.remove(max(set_nums))
            set_nums.remove(max(set_nums))
            return max(set_nums)
"""
将数组转为set集合去重,再判断集合的长度,若长度小于3,则返回集合的最大值,
若集合长度大于等于3,即原数组存在第三大的元素,则连续两次删除集合最大元素,
得到的集合的最大值即为原数组的第三大元素。时间复杂度为O(n)
"""