Leetcode 23-合并K个升序链表

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

请你将所有链表合并到一个升序链表中,返回合并后的链表。
在这里插入图片描述

题解

构建一个大小为k的小根堆,堆中初始存放每个链表的头结点
当堆不为空,每次从堆中poll出值最小的链表节点接到新链表上
当被poll出节点的下一个节点不为空,把被poll出的链表节点的下一个节点入堆
返回新链表头结点

/**
 * 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) {
        //构建小根堆,因为存储的是node节点,所以需要重写比较方法
        Queue<ListNode> pq = new PriorityQueue<>((v1,v2)->v1.val-v2.val);
        //将数组中每个链表的头结点放入堆中
        for(ListNode node:lists){
            if(node!=null){
                pq.offer(node);
            } 
        }
        
        //新链表的头结点
        ListNode dummyHead = new ListNode(0);
        ListNode tail = dummyHead;
        while(!pq.isEmpty()){
            //获取当前的值最小的链表节点
            ListNode minNode = pq.poll();
            //将值最小的
            tail.next = minNode;
            tail=minNode;
            //将被放入新链表的节点的下一个节点入堆
            if(minNode.next!=null){
                pq.offer(minNode.next);
            }
        }
        return dummyHead.next;
    }
}

在这里插入图片描述

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值