[LeetCode] 215. Kth Largest Element in an Array

원숭2·2022년 2월 15일
0

LeetCode

목록 보기
47/51

문제

풀이

  1. 최대힙을 사용하기 위해 map함수와 lambda식을 이용해서 기존 값들을 전부 음수로 변환해 줌.
  2. 목표값-1 까지 heappop 한 후, 마지막 heappop값에 -를 붙여서 return함.

코드

import heapq

class Solution:
    def findKthLargest(self, nums: List[int], k: int) -> int:
        nums = list(map(lambda x : -x, nums))
        heapq.heapify(nums)
        
        for _ in range(k-1) :
            heapq.heappop(nums)
        
        return -(heapq.heappop(nums))

0개의 댓글