给你一个链表数组,每个链表都已经按升序排列。
你将所有链表合并到一个升序链表中,返回合并后的链表。
leetcode
解题思路:利用一个优先队列,对有序链表的头节点进行维护,每次取出一个队列头,就能取到节点中的最小值,然后添加到新建的链表尾部,然后如果这个节点有next,就把他的next再放进优先队列中。直到队列取完。
class Solution {
public ListNode mergeKLists(ListNode[] lists) {
ListNode dummy = new ListNode(0);
ListNode cur = dummy;
PriorityQueue<ListNode> q = new PriorityQueue<>((a, b) -> a.val - b.val);
for(ListNode l : lists) {
if(l != null) q.offer(l);
}
while(!q.isEmpty()) {
ListNode temp = q.poll();
cur.next = temp;
if(temp.next != null) q.offer(temp.next);
cur = cur.next;
}
cur.next = null;
return dummy.next;
}
}