Explain merging k sorted lists.
01
01
Understand the problem
heaplinked-list
02
02
Attempt it yourself
Sketch your approach before reading the solution — that's what interviews test.
Stuck? AI Nudge Available
Get a conceptual hint to guide your logic without spoiling the final implementation.
03
03
Study the solution
The solution is waiting
Give it an honest attempt first — then compare your thinking with the full walkthrough.
04
04
Read the code
Min-heap of list heads
Run Playgroundimport heapq
class ListNode:
def __init__(self, val, nxt=None):
self.val, self.next = val, nxt
def merge_k_lists(lists):
heap = []
for i, node in enumerate(lists):
if node:
heapq.heappush(heap, (node.val, i, node)) # i breaks ties
dummy = tail = ListNode(0)
while heap:
val, i, node = heapq.heappop(heap)
tail.next = node
tail = node
if node.next:
heapq.heappush(heap, (node.next.val, i, node.next))
return dummy.next
# --- demo ---
def build(vals):
head = None
for v in reversed(vals): head = ListNode(v, head)
return head
def to_list(head):
out = []
while head: out.append(head.val); head = head.next
return out
merged = merge_k_lists([build([1, 4, 5]), build([1, 3, 4]), build([2, 6])])
print(to_list(merged)) # [1, 1, 2, 3, 4, 4, 5, 6]05
05
Join the discussion
Discussion (0)
Sign in to join the discussion.
No responses yet. Be the first to share what you think.