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)
"""