设计一个找到数据流中第K大元素的类(class)。注意是排序后的第K大元素,不是第K个不同的元素。
你的KthLargest类需要一个同时接收整数k 和整数数组nums的构造器,它包含数据流中的初始元素。每次调用KthLargest.add,返回当前数据流中第K大的元素。
示例:
思路:使用最小堆,最小堆的大小是K,所以第K大的值就是堆顶那个元素。每次进来新的元素和堆顶比较,如果比对顶小,那第K大还是原来堆顶的元素;如果新元素比堆顶大,那就把堆顶的元素移除,新的元素加入后重新排序,返回堆顶元素。