Leetcode 703: 求一个stream中第k大的数字

想法：
我们可以用堆队列算法完成这个任务。可以参考这个网页：堆队列算法。

题目要求我们找到第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)