描述
给你一个链表数组,每个链表都已经按升序排列。
请你将所有链表合并到一个升序链表中,返回合并后的链表。
示例 1:
输入: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
示例 2:
输入:lists = []
输出:[]
说明
队列是一种特征为FIFO的数据结构,每次从队列中取出的是最早加入队列中的元素。但是,许多应用需要另一种队列,每次从队列中取出的应是具有最高优先权的元素,这种队列就是优先级队列(Priority Queue),也称为优先权队列。
代码
/**
* 合并K个有序链表
*
* @param lists 链表数组
* @return 合并后的链表
*/
public ListNode mergeKLists(ListNode[] lists) {
if (lists == null || lists.length == 0) return null;//空值直接返回
//实例化优先级队列
PriorityQueue<ListNode> queue = new PriorityQueue<>(lists.length, Comparator.comparingInt(o -> o.val));
ListNode result = new ListNode(0);//用于保存结果
ListNode temp = result;//临时变量,temp引用对象指向result
for (ListNode node : lists) {//链表入队列
if (node != null) queue.add(node);
}
while (!queue.isEmpty()) {
temp.next = queue.poll();//优先级最大的出队列,即队列中头节点最小的链表
temp = temp.next;//移动链表指针
if (temp.next != null) queue.add(temp.next);//将移动后的链表重新入队列
}
return result.next;
}
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;
}
}