코딩테스트 문제 (10) - 배열의 K 번째 큰 요소

제목 그대로 주어진 배열이 있을 때, 배열의 K 번째 큰 요소를 return하면 되는 문제입니다.

문제

1. array=[3,2,3,1,2,4,5,5,6], K=4
=> 4

풀이

주어진 배열의 우선순위를 기반으로 가장 큰 값부터 K번 추출하면 되는 간단한 문제입니다. 힙을 사용해 쉽게 구현할 수 있습니다.

kaggle.com의 notebook으로 작성한 풀이는 다음과 같습니다.

홍머스 정리

• 힙, heapq
• 난이도: 하

참조

• <파이썬 알고리즘 인터뷰>, p456-457