leetcode-第三大的数

414. 第三大的数

暴力解法：

先去重，然后降序排序，去重排序后的数组的长度大于等于3的时候，取下标为2的即为结果值，当去重后的数组长度小于3，取最大值即可

class Solution:
    def thirdMax(self, nums: List[int]) -> int:
        nums = list(set(nums))
        nums.sort(reverse=True)
        if len(nums) >= 3:
            return nums[2]
        else:
            return max(nums)

题目中还有一个要求：

进阶：你能设计一个时间复杂度 O(n) 的解决方案吗？

使用堆（特别是最小堆）来解决这个问题，可以帮助我们保持跟踪数组中最大的三个数。在Python中，我们可以使用标准库heapq里的最小堆实现。关键思路是维护一个大小为3的最小堆，这样堆顶始终是这个堆中最小的数，当堆的大小超过3时，我们可以弹出最小的数，最后堆中剩下的就是数组中最大的三个数。如果堆的大小小于3，说明不存在第三大的数，则返回堆中的最大数。

import heapq

class Solution:
    def thirdMax(self, nums: List[int]) -> int:
        heap = []
        nums = set(nums)
        for num in nums:
            heapq.heappush(heap, num)
            if len(heap) > 3:
                heapq.heappop(heap)
        return heap[0] if len(heap) == 3 else max(heap)