‘가장 큰 K개’은 힙의 핵심 연산을 다른 입력 조건에서도 스스로 선택할 수 있게 만드는 반복 문제입니다. 결과뿐 아니라 경계값과 목표 복잡도까지 확인하세요.
heapq를 사용해 가장 큰 k개를 내림차순으로 반환하세요.
k_largest(numbers, k)
시간 O(n) 이내
실행하면 아래 순서대로 채점합니다. print() 출력도 이 순서대로 쌓입니다.
| # | 이름 | numbers | k | 기대값 |
|---|---|---|---|---|
| 1 | 두 개 | [4, 1, 7, 3] | 2 | [7, 4] |
| 2 | 큰 k | [2, 1] | 5 | [2, 1] |
| 3 | 0 | [1] | 0 | [] |
| 4 | 중복 최댓값 | [5, 1, 5, 3] | 3 | [5, 5, 3] |
| 5 | 음수만 | [-8, -2, -5, -1] | 2 | [-1, -2] |
Python 표준 라이브러리는 사용할 수 있습니다. 함수 이름과 매개변수는 제시된 형태를 유지하세요.
heapq는 최소 힙입니다.코드를 작성하고 실행하면 5개의 테스트가 각각 표시됩니다.