#!/usr/bin/env python3 # -*- coding: utf-8 -*- # author: bt3gl class KthLargest: def __init__(self, k, nums): self.k = k self.heap = nums heapq.heapify(self.heap) while len(self.heap) > k: heapq.heappop(self.heap) def add(self, val: int) -> int: heapq.heappush(self.heap, val) if len(self.heap) > self.k: heapq.heappop(self.heap) return self.heap[0]