전체 정렬이 필요하지 않고 최솟값 몇 개만 필요할 때 힙을 선택하는 법을 익힙니다. 우선순위 큐가 어떤 문제에서 정렬보다 적합한지 판단하는 연습입니다.
정수 리스트에서 가장 작은 k개를 오름차순 리스트로 반환하세요. 입력은 변경하지 않으며 sorted()와 sort()는 사용하지 않습니다. heapq를 사용하세요.
k_smallest(numbers, k)
힙 구성 O(n), k회 추출 O(k log n)
실행하면 아래 순서대로 채점합니다. print() 출력도 이 순서대로 쌓입니다.
| # | 이름 | numbers | k | 기대값 |
|---|---|---|---|---|
| 1 | 일반 | [7, 2, 9, 1, 5] | 3 | [1, 2, 5] |
| 2 | 중복 | [4, 1, 1, 3] | 2 | [1, 1] |
| 3 | k가 더 큼 | [3, 2] | 5 | [2, 3] |
| 4 | k가 0 | [1, 2] | 0 | [] |
| 5 | 음수 포함 | [4, -3, 8, -1, 2] | 3 | [-3, -1, 2] |
Python 표준 라이브러리는 사용할 수 있습니다. 함수 이름과 매개변수는 제시된 형태를 유지하세요.
heapq는 최소 힙입니다.코드를 작성하고 실행하면 5개의 테스트가 각각 표시됩니다.