《算法通关村------解析堆在合并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 = [[]]
输出:[]
这里直接给代码,我们来理解
java
/**
* Definition for singly-linked list.
* public class ListNode {
* int val;
* ListNode next;
* ListNode() {}
* ListNode(int val) { this.val = val; }
* ListNode(int val, ListNode next) { this.val = val; this.next = next; }
* }
*/
class Solution {
public ListNode mergeKLists(ListNode[] lists) {
if(lists == null || lists.length == 0){
return null;
}
PriorityQueue<ListNode> q = new PriorityQueue<>(Comparator.comparing(node -> node.val));
/*
理解下面这个循环,只是把每个链表的第一个数放入最小堆中。
*/
for(int i = 0;i< lists.length;i++){
if(lists[i] != null){
q.add(lists[i]);
}
}
/*
理解下面这个循环,首先就是把上面的每个链表的第一个元素进行了一个排序,然后把那个最小的元素的链表拿出来,把它放入我们定义的新的链表的下一个节点,判断他是否有下一个节点,如果有下一个节点,那么就把他放入最小堆里面去(此时堆会重新排序),然后再进行下一次循环,从最小堆中取元素。其实理解这里最重要就是要知道链表初始就是排好序的。
*/
ListNode dummy = new ListNode(0);
ListNode tail = dummy;
while(!q.isEmpty()){
tail.next = q.poll();
tail = tail.next;
if(tail.next != null){
q.add(tail.next);
}
}
return dummy.next;
}
}
画图理解: