问题:
合并 k 个排序链表,返回合并后的排序链表。请分析和描述算法的复杂度。
示例:
输入:
[
1->4->5,
1->3->4,
2->6
]
输出: 1->1->2->3->4->4->5->6
题解:
做这个题的时候,我们肯定做过合并两个排序链表噻,代码下面贴出来了的。这个题其实就是变向的做合并2个排序链表,只是看你怎么个合法,你可以用前一个链表融合在第二个,然后第二个合并到第三个,最后直到最后一个为止。只不过这样有一个问题,通过这种方式,链表都遍历了多次,所以肯定要费时一些,那么如何能做到一趟下来减少遍历次数。于是俩俩组合的方法出现了,如图,对链表俩俩分组,然后合并,不断分组合并,直到最后只剩一个链表为止。当然,有个小问题就是链表的个数为基数,我们可以将前面偶数个合并,最后一个直接和第一个合并呗!这样最后的答案就是lists[0]了。
如何所示:通过这种合并的方式,我们的链表集合就缩短为绿色部分,然后一直合并,这样以O(1)的空间负责度实现合并,O(nlog(k))时间复杂度。
最后给出代码:
public ListNode mergeKLists(ListNode[] lists) {
if (lists == null || lists.length == 0){
return null;
}
int l = lists.length;
while(l>1){
int d = l/2,i;
for (i = 0; i < d; i++) {
lists[i] = merge2Lists(lists[i],lists[i+d]);
}
if ((l&1)==1){
lists[0] = merge2Lists(lists[0],lists[l-1]);
}
l=l/2;
}
return lists[0];
}
//合并两个排序的链表,采用递归的方式,空间复杂度O(1),这也是leetcode21题答案
public ListNode merge2Lists(ListNode l1,ListNode l2){
if (l1 == null){
return l2;
}else if (l2 == null){
return l1;
}else if (l1.val < l2.val){
l1.next = merge2Lists(l1.next,l2);
return l1;
}else{
l2.next = merge2Lists(l1,l2.next);
return l2;
}
}
```