题解 | 合并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; } }