‘K개 정렬 리스트 병합’은 힙의 핵심 연산을 다른 입력 조건에서도 스스로 선택할 수 있게 만드는 반복 문제입니다. 결과뿐 아니라 경계값과 목표 복잡도까지 확인하세요.
K개 정렬 리스트를 최소 힙으로 하나의 정렬 리스트로 합치세요.
merge_k_sorted(lists)
시간 O(N log K), 공간 O(K)
실행하면 아래 순서대로 채점합니다. print() 출력도 이 순서대로 쌓입니다.
| # | 이름 | lists | 기대값 |
|---|---|---|---|
| 1 | 세 목록 | [[1, 4], [2, 5], [3, 6]] | [1, 2, 3, 4, 5, 6] |
| 2 | 빈 목록 포함 | [[], [1]] | [1] |
| 3 | 모두 빈 목록 | [[], []] | [] |
| 4 | 중복과 음수 | [[-3, 1, 1], [-2, 1, 4], [-3, 5]] | [-3, -3, -2, 1, 1, 1, 4, 5] |
| 5 | 한 긴 목록 | [[0, 2, 4, 6]] | [0, 2, 4, 6] |
Python 표준 라이브러리는 사용할 수 있습니다. 함수 이름과 매개변수는 제시된 형태를 유지하세요.
heapq는 최소 힙입니다.코드를 작성하고 실행하면 5개의 테스트가 각각 표시됩니다.