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

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))

?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發布,文章內容僅代表作者本人觀點,簡書系信息發布平臺,僅提供信息存儲服務。