215. 数组中的第K个最大元素

1. 题目链接
2. 解题思路：使用优先级队列，大小是K，小根堆。遍历数组，当前数为num，如果num小于堆顶，那么num不可能是结果，如果num大于堆顶，那么就把堆顶元素弹出去，同时把num放入优先级队列，依次遍历完数组，堆顶元素就是答案。

• • 为什么正确？
  • 因为优先级队列保持着【当前最大的k个元素】，而堆顶则是【门槛】，遍历一个新的数，如果连【门槛】都达不到，肯定不是答案。遍历完数组，更新优先级队列后，优先级队列就是【全局最大的k个元素】

3. 代码

   class Solution:
       def findKthLargest(self, nums: List[int], k: int) -> int:
           heap = []
           for num in nums:
               if len(heap) < k:
                   heapq.heappush(heap, num)
               elif heap[0] < num:
                   heapq.heappop(heap)
                   heapq.heappush(heap, num)
           return heap[0]