Leetcode--Java--23. 合并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


/**
 * 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; }
 * }
 */

思路

1. 如何同时比较k个链表的结点，然后选取其中最小的？ 考虑用堆，在java中就用优先队列PriorityQueue来实现
2. 优先队列不能为空。设置好比较顺序为升序，比较的是结点的值。这样把所有结点放进去，每次取出的就是最小的那个结点
3. 取出的时候要用尾插法构建合并后的链表，因为尾插法和插入顺序是一致的
4. lamda表达式(a, b）-> (a.val - b. val) 表示升序
5. offer() 加入元素，poll()取出第一个元素并删除，peek() 查看/取出第一个元素，但不删除
6. 链表题一般都要设置虚拟头结点，使得对头结点的操作与对其他结点的操作保持一致

代码

class Solution {
    public ListNode mergeKLists(ListNode[] lists) {
       if (lists == null) return null;
       ListNode dummy = new ListNode(-1);
       ListNode tail = dummy;
       //建立升序的优先队列   按照结点值排序
       PriorityQueue<ListNode> pq = new PriorityQueue<>((a, b) -> (a.val - b.val));
       //将所有链表的头结点加入到 优先队列
       for (int i = 0; i < lists.length; i ++ ) {
           if (lists[i] != null) pq.offer(lists[i]);
       }
       //组合链表
       while (!pq.isEmpty()) {
           ListNode t = pq.poll();
           //利用尾插法构建组合链表，因为尾插法与插入顺序一致  都是升序
           tail.next = t;
           tail = t;
           //如果后面还有结点，就加入   因为可能第一个链表的都偏小，一次加入多个
           if (t.next != null) pq.offer(t.next);
       }
       return dummy.next;
    }
}