414. 第三大的数
题目描述
给你一个非空数组,返回此数组中 第三大的数 。如果不存在,则返回数组中最大的数。
示例 1:
输入:[3, 2, 1]
输出:1
解释:第三大的数是 1 。
示例 2:
输入:[1, 2]
输出:2
解释:第三大的数不存在, 所以返回最大的数 2 。
示例 3:
输入:[2, 2, 3, 1]
输出:1
解释:注意,要求返回第三大的数,是指在所有不同数字中排第三大的数。
此例中存在两个值为 2 的数,它们都排第二。在所有不同数字中排第三大的数为 1 。
提示:
1 <= nums.length <= 104
-231 <= nums[i] <= 231 - 1
解题:
最大堆的典型用法。最大堆可以在 O ( l o g 2 N O(log_2N O(log2N)的时间复杂度内完成对插入的数的排序。
在本题中,因为我们只需维护容量为3的最大度,所以维护的时间为常数。总的时间复杂度就是遍历数组的时间复杂度 O ( N ) O(N) O(N) 。
在python代码中,这里用有序列表取代了堆的复杂操作(又是奇怪python库的奇怪用法,虽然很香)。
from sortedcontainers import SortedList
class Solution:
def thirdMax(self, nums: List[int]) -> int:
s = SortedList()
for num in nums:
if num not in s:
s.add(num)
if len(s) > 3:
s.pop(0)
return s[0] if len(s) == 3 else s[-1]