题目链接:合并K个排序链表
题目描述:
合并 k 个排序链表,返回合并后的排序链表。请分析和描述算法的复杂度。
示例:
输入:
[
1->4->5,
1->3->4,
2->6
]
输出: 1->1->2->3->4->4->5->6
题目分析:基于合并2个有序链表的过程,合并K个链表,最简单的方式就是,将K个链表进行K-1次合并(每次合并2个,并保留上一次合并的结果进行下一次合并),这个时间复杂度是O(n2),寻求更低的时间复杂度是我们不断追求的目标。合并K个链表的过程是不是觉得似曾相识呢?没错,就是归并排序,这个基本上就是归并排序了,每次进行两两归并,归并的过程就是组合链表的过程,时间复杂度是O(nlogn)
代码:
class Solution {
public ListNode mergeKLists(ListNode[] lists) {
return mergeLists(lists,0,lists.length-1);
}
public ListNode mergeLists(ListNode[] st,int l,int r){
//分组只有一个元素了,开始归并
if(r-l==0) return st[l];
if(l>r) return null;
int m = l+(r-l)/2;
ListNode a,b,head = new ListNode(-1);
//区间为左闭右闭
a = mergeLists(st,l,m);
b = mergeLists(st,m+1,r);
ListNode p = head;
//
while(a!=null&&b!=null){
if(a.val<b.val){
p.next = a;
a = a.next;
}else{
p.next = b;
b = b.next;
}
p = p.next;
}
p.next = a==null?b:a;
return head.next;
}
}
继续思考:仔细观察一下合并的过程,可以发现,每次都是选取的K个链表中第一个元素的最小值作为新链表的节点,那么可以使用一个大小为K的优先队列来维护这些值,新链表的值只需要不断从队列取就可以了
class Solution {
public ListNode mergeKLists(ListNode[] lists) {
PriorityQueue<ListNode> q = new PriorityQueue<>(Comparator.comparingInt(o -> o.val));
for (ListNode head: lists) {
if(head!=null) q.add(head);
}
ListNode head = new ListNode(-1);
ListNode p = head;
while(!q.isEmpty()){
//取最小的
ListNode te = q.poll();
p.next = te;
p = p.next;
if(te.next!=null) q.add(te.next);//继续加入新的第一个节点
}
return head.next;
}
}
不过这个执行时间好像更长了耶。。。,体会思路,思路哈😁