LeetCode 高级 - 合并K个元素的有序链表

合并K个元素的有序链表

合并 k 个排序链表,返回合并后的排序链表。请分析和描述算法的复杂度。

示例:

输入:
[
  1->4->5,
  1->3->4,
  2->6
]
输出: 1->1->2->3->4->4->5->6

分析

采用最小堆的思想,将K个链表的头节点放入优先队列中,遍历时返回队首元素,并把该队首元素的下一节点放入优先队列中,直至优先队列为空。

注意:该题有坑,K个链表中,可能有空链表!!

代码

    /**
     * Definition for singly-linked list.
     * public class ListNode {
     *     int val;
     *     ListNode next;
     *     ListNode(int x) { val = x; }
     * }
     */
    class Solution {
        //思路是 优先队列(小顶堆) 遍历

        //遇到的坑!!!K个节点,每个节点都有可能是空!!!也就是说虽然lists数有长度,但是可能很多个元素都是null
        public ListNode mergeKLists(ListNode[] lists) {
            PriorityQueue<ListNode> queue = new PriorityQueue<>((a,b)->a.val-b.val);
            for(int i=0;i<lists.length;i++){
                if(lists[i]!=null){
                    queue.add(lists[i]);
                }
            }
            if(queue.isEmpty()){
                return null;
            }
            ListNode temp = new ListNode(queue.peek().val);
            ListNode head = temp;
            while(!queue.isEmpty()){
                ListNode p = queue.poll();
                temp.next = p;
                temp = temp.next;
                if(p.next!=null){
                    queue.offer(p.next);
                }
            }
            return head.next;
        }
    }
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值