题解 | 合并k个已排序的链表
合并k个已排序的链表
https://www.nowcoder.com/practice/65cfde9e5b9b4cf2b6bafa5f3ef33fa6
import java.util.*;
/*
* public class ListNode {
* int val;
* ListNode next = null;
* public ListNode(int val) {
* this.val = val;
* }
* }
*/
public class Solution {
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
*
* @param lists ListNode类ArrayList
* @return ListNode类
*/
public ListNode mergeKLists (ArrayList<ListNode> lists) {
// write code here
PriorityQueue<ListNode> queue = new PriorityQueue<>((a,b) -> a.val - b.val);
for(ListNode node :lists){
if(node != null){
queue.add(node);
}
}
ListNode dummy = new ListNode(0);
ListNode tail = dummy;
// 当堆不为空时,取出最小元素,并将其下一个节点加入到堆中
while(!queue.isEmpty()){
ListNode node = queue.poll();
tail.next = node;
tail = tail.next;
if (node.next != null){
queue.add(node.next);
}
}
return dummy.next;
}
}

字节跳动公司福利 1374人发布