File tree Expand file tree Collapse file tree 1 file changed +13
-6
lines changed Expand file tree Collapse file tree 1 file changed +13
-6
lines changed Original file line number Diff line number Diff line change 1
1
"""
2
2
https://leetcode.com/problems/merge-k-sorted-lists/description/
3
3
4
- 문제: k개의 링크드 리스트가 주어지고, 각 링크드 리스트가 오름차순으로 정렬이 되어있다. 모든 링크드 리스트를 병합하여 하나의 정렬된 링크드 리스트를 만들어라.
4
+ 문제: k개의 링크드 리스트가 주어지고, 각 링크드 리스트가 오름차순으로 정렬이 되어있다.
5
+ 모든 링크드 리스트를 병합하여 하나의 정렬된 링크드 리스트를 만들어라.
5
6
6
7
풀이:
7
8
heapq를 쓰면 항상 가장 작은 값을 O(log k) 시간에 꺼낼 수 있음.
8
- heapq -> 최소 힙 구조 -> 내부적으로 항상 가장 작은 값이 루트에 오도록 정렬됨.
9
-
9
+ heapq -> ' 최소 힙 구조' -> 내부적으로 항상 가장 작은 값이 루트에 오도록 정렬됨.
10
+
10
11
1. 각 리스트의 첫 노드를 heap에 넣음 (val, 고유번호, 노드)
11
12
2. heap에서 가장 작은 값 꺼내면서 결과 리스트 구성
12
13
3. 다음 노드를 힙에 추가
13
14
14
- TC: O(n log k), SC: O(k)
15
- n = 모든 리스트의 노드 수
16
- k = 리스트의 개수
15
+ n = 모든 노드의 총 개수
16
+ k = 연결 리스트의 개수
17
+
18
+ TC: O(n log k)
19
+ - n개의 노드를 각각 힙에 1번씩 push, 1번씩 pop함 → 총 2n번 힙 연산
20
+ - 각 힙 연산은 log k 시간 (힙 크기 최대 k)
21
+
22
+ SC: O(k)
23
+ - 힙에 최대 k개의 노드가 동시에 들어감
17
24
"""
18
25
19
26
from typing import List , Optional
You can’t perform that action at this time.
0 commit comments