classKthLargest:def__init__(self,k,nums):""" :type k: int :type nums: List[int] """ self.nums = nums self.k = k heapq.heapify(self.nums)whilelen(self.nums)> k: heapq.heappop(self.nums)defadd(self,val):""" :type val: int :rtype: int """ heapq.heappush(self.nums, val)whilelen(self.nums)> self.k: heapq.heappop(self.nums)return self.nums[0]# Your KthLargest object will be instantiated and called as such:# obj = KthLargest(k, nums)# param_1 = obj.add(val)