2023-2-7 合并K个升序链表【链表】

题目

给你一个链表数组，每个链表都已经按升序排列。请你将所有链表合并到一个升序链表中，返回合并后的链表。

思路：
这里需要用到了解堆的数据结构，链接 https://blog.csdn.net/qq_45399396/article/details/114237345
采用最小堆：可以直接使用PriorityQueue这个线程的优先级队列。思路还是很简单的，我们将所有链表头节点都放进去，然后开始循环出堆，此时出来的元素就是最小的一个元素，然后将出堆节点的next再次加入到最小堆中即可。

代码如下

public ListNode mergeKLists( ListNode[] lists ){
    
    

   if( lists == null || lists.length == 0 ){
    
    
      return null;
   }
   //构建最小堆
   PriorityQueue<ListNode> queue = new PriorityQueue<>( ( o1, o2 ) -> o1.val - o2.val );
   ListNode dummy = new ListNode( 0 );
   ListNode p = dummy;
   //所有头节点加入堆
   for( ListNode node : lists ){
    
    
      if( node != null ) queue.add( node );
   }
   //堆排序
   while( !queue.isEmpty() ){
    
    
      //取出最小节点
      p.next = queue.poll();
      p = p.next;
      //最小节点的后继不为null，则将后继加入最小堆
      if( p.next != null ){
    
    
         queue.add( p.next );
      }
   }
   return dummy.next;
}