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
示例 2:
输入:lists = []
输出:[]
示例 3:
输入:lists = [[]]
输出:[]

最重要的就是使用这个PriorityQueue优先级队列,这种队列可以保证出队列如果是整型的一定是出最小值,由于数据不是整型而是ListNode类型,所以说需要定义一个比较器Comparator,并且实现compare方法

第二步就是把所有的头结点全部放入队列,这样出队列先出的一定是val值最小的头结点,先把这个值出队列,并且把这个值插入到哨兵temp的后面,接着哨兵往后移一位:temp=temp.next
接着就把这个头结点的Next再次放入队列,优先级队列会把这个数据放到应该放的位置,这样一直循环直到队列为空即可。

class Solution {
    public ListNode mergeKLists(ListNode[] lists) {
        PriorityQueue<ListNode> pq = new PriorityQueue(new Comparator<ListNode>(){
            public int compare(ListNode l1,ListNode l2){
                if(l1.val-l2.val<0){
                    return -1;
                }
                else if(l1.val-l2.val>0){
                    return 1;
                }else{
                    return 0;
                }

            }
        });
        for(ListNode s:lists){
            if(s!=null){
                pq.offer(s);
            }
            
        }
        ListNode temp = new ListNode(-1);
        ListNode h = temp;

        while(!pq.isEmpty()){
            ListNode list=pq.poll();
            temp.next=list;
            temp=temp.next;
            if(list.next!=null){
                pq.offer(list.next);
            }
        }
        return h.next;
        
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值