合并多个有序链表
1.问题描述
合并 k 个排序链表,返回合并后的排序链表。请分析和描述算法的复杂度。
示例:
输入:
A [
1->4->5,
1->3->4,
2->6
]
输出: 1->1->2->3->4->4->5->6
2.算法
我起先的想法是先判断链表长度,如果链表长度K>3,则将链表A[1~K-1]的剩余链表当成一个新的链表数组进行递归合并,直至最后两条链表,即A[K-1]和A[K-2]进行合并,然后返回一条合并完的新链表,但是这样的话占用的空间是(K+K-1+K-2+…2+1)*节点空间的大小,占用的空间太大,在leetcode上无法通过。
后面我转换了一下思路,先将A中的链表两两合并,并将合并完的新链表放到一个新的数组B中,如果A的长度K是个奇数,则在最后将A的最后一条链表放入B中,然后将B进行递归合并,直到合并完后的链表只剩下一条。这样占用的空间就只有(K+K/2+K/(2²)+…+1)大小即空间复杂度为O(logk),占用空间下降了很大,同时这个算法的复杂度也是 O(kn×logk),。具体代码如下:
/**
* 链表节点定义
* public class ListNode {
* int val;
* ListNode next;
* ListNode(int x) { val = x; }
* }
*/
//合并K个有序链表
public class Solution {
public ListNode mergeKLists(ListNode[] lists) {
//判断特殊条件链表
if(lists.length==0)
return null;
//判断是否只有一条链表
if(lists.length<2)
return lists[0]