【LeetCode】数组中的第K个最大元素

---

一、题目

给定整数数组 nums 和整数 k，请返回数组中第 k 个最大的元素。

请注意，你需要找的是数组排序后的第 k 个最大的元素，而不是第 k 个不同的元素。

你必须设计并实现时间复杂度为 O(n) 的算法解决此问题。

示例 1:

输入: [3,2,1,5,6,4], k = 2
输出: 5
示例 2:

输入: [3,2,3,1,2,4,5,5,6], k = 4
输出: 4

提示：

1 <= k <= nums.length <= 10 ^5
-10 ^4 <= nums[i] <= 10 ^4

二、解法

大根堆：堆顶元素是堆中最大的
小根堆：堆顶元素是堆中最小的
可以利用大根堆，首先将所有元素放入堆中，然后弹出堆顶元素k-1次，弹出之后，最后堆顶剩下的元素就是最大的了
但注意，python里面的堆默认是小根堆，所以对于x，我们可以将-x放入，取的时候取-x，就可以得到逻辑意义上的大根堆了。

---

完整代码

class Solution:
    def findKthLargest(self, nums: List[int], k: int) -> int:
        hp = []
        heapify(hp)
        for x in nums:
            heappush(hp,-x)
        for _ in range(k - 1):
            heappop(hp)
        return -hp[0]