合并链表-leetcode23-合并k个升序链表-526互联

给你一个链表数组，每个链表都已经按升序排列。

请你将所有链表合并到一个升序链表中，返回合并后的链表。

示例 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

示例 2：

输入：lists = []
输出：[]

示例 3：

输入：lists = [[]]
输出：[]

提示：

k == lists.length
0 <= k <= 10^4
0 <= lists[i].length <= 500
-10^4 <= lists[i][j] <= 10^4
lists[i] 按升序排列
lists[i].length 的总和不超过 10^4

思路

对于多个数组要合并成一条链表，首先就是把他们排好序，可以借助优先级队列，其次就是把他们取出来，由于放入优先级队列的只是每个数组的头节点，所以排序只是针对这三个数，每次取出一个节点放入新节点后，如果这个节点后面还有节点也需要放入队列中进行排序，依次如这般操作即可组成新链表

//leetcode submit region begin(Prohibit modification and deletion)

/**
 * 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; }
 * }
 */
class Solution {
    public ListNode mergeKLists(ListNode[] lists) {

        if(lists==null || lists.length==0){
            return null;
        }

        ListNode dummy = new ListNode(-1);
        ListNode p = dummy;


        //利用优先级队列，大数都沉下去
        PriorityQueue<ListNode> pq = new PriorityQueue<>(lists.length, (a, b) -> (a.val - b.val));
        //PriorityQueue<ListNode> queue = new PriorityQueue<>(
        //        lists.length, (a, b)->(a.val - b.val));



        //放数 放都是每个数组的头节点
        for (ListNode head : lists) {
            if(head!=null){
                pq.add(head);
                //System.out.println(("pq " + head.val));
            }
        }



        //取数构成链表，注意每次取的都是每个链表的头节点，取完记得把当前链表的下一个节点放进去
        while (pq != null && pq.size() != 0) {
            ListNode tempNode = pq.poll();
            p.next = tempNode;
            p = p.next;

            if(tempNode!=null&&tempNode.next!=null){
                pq.add(tempNode.next);
            }

            //System.out.println("p: " + p.val);
        }
        return dummy.next;
    }
}
//leetcode submit region end(Prohibit modification and deletion)