[LeetCode] 215. Kth Largest Element in an Array (Python)

• 최대힙 개념 사용 (마이너스)
• element 값을 리턴하는 건데 순간 index의 개념으로 착각했음
  • answer를 +1씩 카운트하다가 조건문을 만났을 때 리턴하는 형식으로 해줬었는데 그렇게 하면 안 되지

Solution

import heapq
from typing import List


class Solution:
    def findKthLargest(self, nums: List[int], k: int) -> int:
        answer = 0

        heap = []
        for num in nums:
            heapq.heappush(heap, -num)

        for _ in range(k):
            answer = -heapq.heappop(heap)

        return answer

Reference

• https://leetcode.com/problems/kth-largest-element-in-an-array