Day74：數據流中的第K大元素

設計一個找到數據流中第K大元素的類（class）。注意是排序后的第K大元素，不是第K個不同的元素。

你的 KthLargest 類需要一個同時接收整數 k 和整數數組nums 的構造器，它包含數據流中的初始元素。每次調用

KthLargest.add，返回當前數據流中第K大的元素。

示例:

int k = 3;

int[] arr = [4,5,8,2];

KthLargest kthLargest = new KthLargest(3, arr);

kthLargest.add(3); // returns 4

kthLargest.add(5); // returns 5

kthLargest.add(10); // returns 5

kthLargest.add(9); // returns 8

kthLargest.add(4); // returns 8

說明:

你可以假設 nums 的長度≥ k-1 且k ≥ 1。

來源：力扣（LeetCode）

鏈接：??力扣…?

著作權歸領扣網絡所有。商業轉載請聯系官方授權，非商業轉載請注明出處。

import heapq
class KthLargest(object):

def __init__(self, k, nums):
    """
    :type k: int
    :type nums: List[int]
    """
    self.k = k
    self.nums = nums
    heapq.heapify(self.nums)
    while len(self.nums) > self.k:
        heapq.heappop(self.nums)

def add(self, val):
    """
    :type val: int
    :rtype: int
    """
    if val > self.nums[0]:
        heapq.heapreplace(self.nums, val)
        return self.nums[0]
    else:
        return self.nums[0]

Your KthLargest object will be instantiated and called as such:

obj = KthLargest(k, nums)

param_1 = obj.add(val)

def test_KthLargest():
k = 3
arr = [4,5,8,2]
kthlargest = KthLargest(k,arr)
assert kthlargest.add(3) == 4
assert kthlargest.add(5) == 5
assert kthlargest.add(10) == 5
assert kthlargest.add(9) == 8
assert kthlargest.add(4) == 8

if name == 'main':
k = 3
arr = [4,5,8,2]
kthlargest = KthLargest(k,arr)
print(kthlargest.add(3))
print(kthlargest.add(5))