题目链接
小根堆合并K个有序链表
描述:
给定一个链表数组,每个链表都已经按升序排列。
请将所有链表合并到一个升序链表中,返回合并后的链表。输入:lists = [[1,4,5],[1,3,4],[2,6]]
输出:[1,1,2,3,4,4,5,6]
解释:链表数组如下:
[
1->4->5,
1->3->4,
2->6
]
将它们合并到一个有序链表中得到。
1->1->2->3->4->4->5->6
public class ListNode {
int val;
ListNode next;
ListNode() {}
ListNode(int val) { this.val = val; }
ListNode(int val, ListNode next) { this.val = val; this.next = next; }
}
题解
小根堆:
堆是一颗完全二叉树,小根堆顾名思义,最小的节点是根节点,并且每颗子树也都满足该性质,在Java中堆是PriorityQueue,可以根据比较器自定义是小根堆还是大根堆
class Solution {
public ListNode mergeKLists(ListNode[] lists) {
PriorityQueue<ListNode> q = new PriorityQueue<>((a, b) -> a.val - b.val);
for (ListNode e: lists) {
if (e != null) q.add(e);
}
if (q.isEmpty()) return null;
ListNode head = q.poll();
ListNode p = head;
if (p.next != null) q.add(p.next);
while (!q.isEmpty()) {
ListNode poll = q.poll();
p.next = poll;
p = poll;
if (poll.next != null) q.add(poll.next);
}
return head;
}
}