題目描述:
合併 k 個排序鏈表,返回合併後的排序鏈表。請分析和描述算法的複雜度。
示例:
輸入:
[
1->4->5,
1->3->4,
2->6
]
輸出: 1->1->2->3->4->4->5->6
代碼實現:
- 遞歸逐一合併,前面我們已經做了合併兩個有序鏈表,現在通過遞歸,把所有鏈表拆分逐一合併即可。
- 時間複雜度:O(kn)
/**
* Definition for singly-linked list.
* function ListNode(val) {
* this.val = val;
* this.next = null;
* }
*/
/**
* @param {ListNode[]} lists
* @return {ListNode}
*/
var mergeKLists = function(lists) {
switch(lists.length) {
case 0:
return null
case 1:
return lists[0]
case 2:
return mergeTwoLists(lists[0], lists[1])
default:
let mid = Math.round(lists.length / 2)
return mergeTwoLists(mergeKLists(lists.slice(0, mid)),
mergeKLists(lists.slice(mid, lists.length)))
}
};
function mergeTwoLists (l1, l2) {
if (l1 === null) {
return l2
}
if (l2 === null) {
return l1
}
if (l1.val < l2.val) {
l1.next = mergeTwoLists(l1.next, l2)
return l1
} else {
l2.next = mergeTwoLists(l1, l2.next)
return l2
}
};