合并链表-leetcode23-合并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

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

posted @ 2023-03-21 10:45  小傻孩丶儿  阅读(13)  评论(0编辑  收藏  举报