LeeCode(堆,链表 )23_合并K个升序链表
题目:
给你一个链表数组,每个链表都已经按升序排列。
请你将所有链表合并到一个升序链表中,返回合并后的链表。
示例 1:
输入:lists = [[1,4,5],[1,3,4],[2,6]]
输出:[1,1,2,3,4,4,5,6]
解释:链表数组如下:
[
1->4->5,
1->3->4,
2->6
]
将它们合并到一个有序链表中得到。
1->1->2->3->4->4->5->6
示例 2:
输入:lists = []
输出:[]
示例 3:
输入:lists = [[]]
输出:[]
提示:
k == lists.length
0 <= k <= 10^4
0 <= lists[i].length <= 500
-10^4 <= lists[i][j] <= 10^4
lists[i] 按 升序 排列
lists[i].length 的总和不超过 10^4
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/merge-k-sorted-lists
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思路:
- 先弄清两个链表合并成一个链表的思路。 首先创建一个head(ListNode类型)保存合并后的链表的头部,head的val和next可以不给值,合并完成后返回head.next即可。
- 然后创建一个tail来记录下一个插入位置的前一个位置,即目前已合并的最后一个ListNode。
- 创建两个Ptr分别为aPtr和bPtr 分别指向链表A和链表B未合并的第一个ListNode。 用while进行循环合并,每次循环 tail.next=aPtr或bPtr(判断val)然后tail和aPtr或bPtr指向下一个。
- 循环结束后判断链表A和链表B是否为空,不为空将剩余的所有元素合并进去。
合并方法1:顺序合并
我们可以想到一种最朴素的方法:用一个变量 ans 来维护以及合并的链表,第 i 次循环把第 i 个链表和 ans 合并,答案保存到 ans 中。
Java代码:
class ListNode {
int val;
ListNode next;
ListNode() {}
ListNode(int val) { this.val = val; }
ListNode(int val, ListNode next) { this.val = val; this.next = next; }
}
public class 合并两个有序链表 {
public ListNode mergeKLists(ListNode[] lists) {
ListNode ans = null;
for(int i=0;i<lists.length;i++){
ans = mergeTwoLists(ans,lists[i]);
}
return ans;
}
public ListNode mergeTwoLists(ListNode a, ListNode b) {
if(a == null || b == null){
return a!=null?a:b;
}
ListNode head = new ListNode(0);
ListNode tail = head,aPtr = a, bPtr = b;
while(aPtr!=null && bPtr!=null){
if(aPtr.val<bPtr.val){
tail.next = aPtr;
aPtr = aPtr.next;
}else{
tail.next = bPtr;
bPtr = bPtr.next;
}
tail = tail.next;
}
tail.next = (aPtr!=null?aPtr:bPtr);
return head.next;
}
}
合并方法2:分治合并
class ListNode {
int val;
ListNode next;
ListNode() {}
ListNode(int val) { this.val = val; }
ListNode(int val, ListNode next) { this.val = val; this.next = next; }
}
public class 合并两个有序链表 {
public ListNode mergeKLists(ListNode[] lists) {
return merge(lists,0,lists.length);
}
public ListNode merge(ListNode[] lists, int l, int r) {
if(l==r)
return lists[l];
if(l>r)
return null;
int mid = (l+r)/2;
return mergeTwoLists(merge(lists,l,mid),merge(lists,mid+1,r));
}
public ListNode mergeTwoLists(ListNode a, ListNode b) {
if(a == null || b == null){
return a!=null?a:b;
}
ListNode head = new ListNode(0);
ListNode tail = head,aPtr = a, bPtr = b;
while(aPtr!=null && bPtr!=null){
if(aPtr.val<bPtr.val){
tail.next = aPtr;
aPtr = aPtr.next;
}else{
tail.next = bPtr;
bPtr = bPtr.next;
}
tail = tail.next;
}
tail.next = (aPtr!=null?aPtr:bPtr);
return head.next;
}
}