문제55: 배열의 K번째 큰 요소

문제

입력

[3,2,3,1,2,4,5,5,6], k = 4

출력

4

풀이

1. 정렬 후 → 추출 = O(nlogn)

return sorted(nums)[-k]

1. heapq 모듈 사용

heapq.heapify(nums)

for _ in range(len(nums) - k):
	heapq.heappop(nums)

return heapq.heappop(nums)

새로운 개념

• heapq는 최소 힙만을 지원한다.

• heapq.heapify(list) → 한번에 힙으로 만들어 줌