题解 | #合并k个已排序的链表#
合并k个已排序的链表
https://www.nowcoder.com/practice/65cfde9e5b9b4cf2b6bafa5f3ef33fa6
step 1:从链表数组的首和尾开始,每次划分从中间开始划分,划分成两半,得到左边n/2个链表和右边n/2个链表。
step 2:继续不断递归划分,直到每部分链表数为1.
step 3:将划分好的相邻两部分链表,按照两个有序链表合并的方式合并,合并好的两部分继续往上合并,直到最终合并成一个链表。
分治法和双指针
return null;
}
if (left == right) {
return lists.get(left);
}
int mid = (left + right) / 2;
return Merge(divideMerge(lists, left, mid), divideMerge(lists, mid + 1, right));
对于这k个链表,就相当于上述合并阶段的k个子问题,需要划分为链表数量更少的子问题,直到每一组合并时是两两合并,然后继续往上合并,这个过程基于递归:
终止条件: 划分的时候直到左右区间相等或左边大于右边。 返回值: 每级返回已经合并好的子问题链表。 本级任务: 对半划分,将划分后的子问题合并成新的链表。
if (pHead1 == null && pHead2 == null)return null;
if (pHead1 != null && pHead2 == null)return pHead1;
if (pHead1 == null && pHead2 != null)return pHead2;
ListNode head = new ListNode(0);
ListNode cur = head;
while (pHead1 != null && pHead2 != null) {
if (pHead1.val < pHead2.val) {
cur.next = pHead1;
cur = cur.next;
pHead1 = pHead1.next;
} else {
cur.next = pHead2;
cur = cur.next;
pHead2 = pHead2.next;
}
}
if (pHead1 != null)cur.next = pHead1;
if (pHead2 != null)cur.next = pHead2;
return head.next;
}
ListNode divideMerge(ArrayList<ListNode> lists, int left, int right) {
if (left > right) {
return null;
}
if (left == right) {
return lists.get(left);
}
int mid = (left + right) / 2;
return Merge(divideMerge(lists, left, mid), divideMerge(lists, mid + 1, right));
}
public ListNode mergeKLists (ArrayList<ListNode> lists) {
// write code here
return divideMerge(lists, 0, lists.size() - 1);
}
参考题解:https://blog.nowcoder.net/n/0a37bd3344eb499e88697dc648f00b62