力扣 215. 数组中的第K个最大元素 python AC

堆

class Solution:
    def findKthLargest(self, nums, k):
        hp = []
        for i in nums:
            heapq.heappush(hp, i)
            if len(hp) > k:
                heapq.heappop(hp)
        return hp[0]

 创建空列表保存前k个最大元素，遍历列表元素并将元素heappush进新列表中，如果列表超过k个元素，heappop去掉最小的元素，最后返回第0个元素（因为是堆，第0个一定是当前最小）