一.问题描述
合并 k 个排序链表,返回合并后的排序链表。请分析和描述算法的复杂度。
示例:
输入: [ 1->4->5, 1->3->4, 2->6 ] 输出: 1->1->2->3->4->4->5->6
二.代码思路
方法一:暴力循环
- 遍历所有链表,拆开所有节点并把值放到一个数组中。
- 将这个数组排序。
- 遍历该数组将排序后的值组成新链表
- 该算法思路简单,十分暴力,但是时间复杂度较高 ,且代码优化只能在排序算法上加以调整,优化空间太小,所以不推荐此算法。
方法二:分治
- 每次两两链表merge。
- 第一轮合并以后,k 个链表被合并成了 k/2个链表。
- 然后是k/4,k/8等等。
- 直到得到一个有序链表。
- 该算法时间复杂度O((m/2)(k/2 + k/4 + k/8 + …)logk),y也就是O(kmlogk),比方法一有常数量上的优化。空间复杂度为O(1)。
三.方法二代码实现(java)
/**
* Definition for singly-linked list.
* public class ListNode {
* int val;
* ListNode next;
* ListNode(int x) { val = x; }
* }
*/
class Solution {
public ListNode mergeKLists(ListNode[] lists) {
if (lists == null || lists.length == 0) return null;
int begin = 0, end = lists.length - 1;
//两层循环实现两两分组
while (begin < end) {
int mid = (begin + end - 1) / 2;
for (int i = 0; i <= mid; i++) {
lists[i] = merge2list(lists[i], lists[end - i]);
}
end = (begin + end) / 2; //改变长度
}
return lists[0];
}
public ListNode merge2list(ListNode l1, ListNode l2) {
if (l1 == null && l2 == null) return null;
if (l1 == null) return l2;
if (l2 == null) return l1;
ListNode dummy = new ListNode(-1), cur = dummy;
while (l1 != null && l2 != null) {
if (l1.val < l2.val) {
cur.next = l1;
cur = l1;
l1 = l1.next;
} else {
cur.next = l2;
cur = l2;
l2 = l2.next;
}
}
if (l1 != null) cur.next = l1;
if (l2 != null) cur.next = l2;
return dummy.next;
}
}