LeetCode-Merge k Sorted Lists

最新推荐文章于 2022-05-22 13:07:29 发布

原创最新推荐文章于 2022-05-22 13:07:29 发布 · 320 阅读

0 ·

CC 4.0 BY-SA版权

leetcode 同时被 3 个专栏收录

220 篇文章

订阅专栏

queue

6 篇文章

订阅专栏

heap

2 篇文章

订阅专栏

本文介绍了一种利用PriorityQueue数据结构来合并K个已排序链表的方法。通过自定义比较器，该算法能高效地找出最小节点并构建最终的排序链表。文章提供了完整的Java代码实现。

min heap用priority que实现注意判断list数组是否为空等

这里的trick是pop一个node之后不用想记录他是哪个list出来的因为可以用next指针找到！！

public class Solution {
    public ListNode mergeKLists(ListNode[] lists) {
        ListNode node = null;
        if ( lists == null || lists.length == 0 )
            return node;
        ListNode dummy = new ListNode ( 0 );
        ListNode cur = dummy;
        PriorityQueue<ListNode> que = new PriorityQueue<ListNode> ( lists.length, new Comparator<ListNode>(){
            @Override
            public int compare ( ListNode node1, ListNode node2 ){
                return node1.val - node2.val;
            }
        });
        for ( ListNode temp : lists ){
            if ( temp != null )
                que.add ( temp );
        }
        while ( !que.isEmpty() ){
            ListNode temp = que.poll();
            cur.next = temp;
            cur = cur.next;
            if ( cur.next != null )
                que.add ( cur.next );
        }
        return dummy.next;
    }
}