想法:
我们可以用堆队列算法完成这个任务。可以参考这个网页:堆队列算法。
题目要求我们找到第k大的数字,我们可以先把heap里面的元素pop到只剩下k个元素,这时最小的元素就是我们要找的第k大的数字。我们建立一个Min heap,最小的元素会在根的位置,我们可以很快速地找到。
代码:
class KthLargest:
def __init__(self, k: int, nums: List[int]):
self.minHeap, self.k = nums, k
#用list建立一个Min heap。
heapq.heapify(self.minHeap)
#把heap pop到只剩下k个元素,这时最小的元素就是第k大的元素。
while len(self.minHeap) > self.k:
heapq.heappop(self.minHeap)
def add(self, val: int) -> int:
#把val加到heap中。
heapq.heappush(self.minHeap, val)
#把heap中的元素减少到k个。
if len(self.minHeap) > self.k:
heapq.heappop(self.minHeap)
#返回根节点,也就是最小的元素。
return self.minHeap[0]
# Your KthLargest object will be instantiated and called as such:
# obj = KthLargest(k, nums)
# param_1 = obj.add(val)