23. 合并K个排序链表
合并 k 个排序链表,返回合并后的排序链表。请分析和描述算法的复杂度。
示例:
输入:
[
1->4->5,
1->3->4,
2->6
]
输出: 1->1->2->3->4->4->5->6
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/merge-k-sorted-lists
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
结果:
方法一:
class Solution {
public ListNode mergeKLists(ListNode[] lists) {
if(lists == null || lists.length == 0){
return null;
}
//创建一个堆,并设置元素排序方式
PriorityQueue<ListNode> queue = new PriorityQueue(new Comparator<ListNode>() {
public int compare(ListNode o1, ListNode o2){
return (o1.val - o2.val);
}
});
for(int i = 0; i < lists.length; i++){
while(lists[i] != null){
queue.add(lists[i]);
lists[i] = lists[i].next;
}
}
ListNode dummy = new ListNode(-1);
ListNode head = dummy;
while(!queue.isEmpty()){
dummy.next = queue.poll();
dummy = dummy.next;
}
dummy.next = null;
return head.next;
}
思路:创建一个小根堆,将需要排序链数组的所有节点都存入小根堆,每次提取堆的顶部(小根堆的顶部是整个堆的最小值)构建新的链即可。
这个方法是看了别人的思路写的。没有了解过java堆这个结构体。
方法二:
class Solution {
public ListNode mergeKLists(ListNode[] lists) {
if(lists==null || lists.length == 0){
return null;
}
ListNode res = lists[0];
for(int i = 1; i < lists.length; i++){
res = merge(res, lists[i]);
}
return res;
}
public ListNode merge(ListNode l1, ListNode l2){
if(l1 == null || l2 == null){
return l1 == null ? l2 : l1;
}
if(l1.val < l2.val){
l1.next = merge(l1.next, l2);
return l1;
}else{
l2.next = merge(l1, l2.next);
return l2;
}
}
}
思路:这方法结合了昨天的合并两条链表。遍历链表数组,一第一条链表为基础,将第一条和后面的逐一合并。