·연습 문제

K개 정렬 리스트 병합

문제 설명PROBLEM

K개 정렬 리스트 병합

왜 이 문제를 푸나요?

‘K개 정렬 리스트 병합’은 힙의 핵심 연산을 다른 입력 조건에서도 스스로 선택할 수 있게 만드는 반복 문제입니다. 결과뿐 아니라 경계값과 목표 복잡도까지 확인하세요.

익혀야 할 개념

  • 우선순위 큐
  • heapq
  • 최솟값 선택
  • O(log n) 갱신

과제

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 표준 라이브러리는 사용할 수 있습니다. 함수 이름과 매개변수는 제시된 형태를 유지하세요.

개념 노트와 실수 포인트

  • 최솟값을 O(1)에 확인하고 O(log n)에 추가·삭제하는 완전이진트리 구조입니다.
  • 파이썬 heapq는 최소 힙입니다.