23. 合并K个升序链表
原题链接
合并 k 个排序链表,返回合并后的排序链表。请分析和描述算法的复杂度。
代码案例:输入:
[
1->4->5,
1->3->4,
2->6
]
输出: 1->1->2->3->4->4->5->6
题解
- 1、一开始先用小根堆存储k个排序链表的头指针,每次操作后用小根堆维护k个链表当前元素最小的指针,并以指针对应的值进行排序
- 2、操作过程中,当小根堆不为空时,堆顶元素即当前k个排序链表当前最小的元素的指针t,将该值加入到dummy链表的后面,并把t指针往后走一位,使得t指针指向的值变大,再加入到小根堆中
/**
* Definition for singly-linked list.
* 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; }
* }
*/
class Solution {
public ListNode mergeKLists(ListNode[] lists) {
PriorityQueue<ListNode> q = new PriorityQueue<ListNode>((x,y) -> (x.val-y.val));
ListNode dummy = new ListNode(-1),tail =dummy;
//每个链表的头节点插入到堆里面去
for(int i = 0; i< lists.length ; i++){
if(lists[i] !=null) q.add(lists[i]);
}
while(!q.isEmpty()){
ListNode t = q.poll();
tail=tail.next = t ;//插入到为节点后面 之后为节点要更新
if(t.next != null) q.add(t.next);
}
return dummy.next;
}
}