LeetCode: 23. Merge k Sorted Lists

最新推荐文章于 2024-04-03 21:24:25 发布

原创最新推荐文章于 2024-04-03 21:24:25 发布 · 133 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#LeetCode #Java

LeetCode 专栏收录该内容

11 篇文章

订阅专栏

本文介绍了一种使用优先队列解决LeetCode第23题“合并K个排序链表”的高效算法。通过定义比较器，优先队列能够有效地选择并合并最小元素，形成最终的排序链表。

题目: 23. Merge k Sorted Lists(https://leetcode.com/problems/merge-k-sorted-lists/description/)

class Solution {
    public ListNode mergeKLists(ListNode[] lists) {
        ListNode result = null, tail = null;
        PriorityQueue<ListNode> queue = new PriorityQueue<>(Comparator.comparingInt(x -> x.val));

        for (ListNode node : lists) {
            if (node == null) {
                continue;
            }
            queue.add(node);
        }
        
        while (!queue.isEmpty()){
            ListNode head = queue.poll();
            if (head.next != null) {
                queue.add(head.next);
            }
            head.next = null;
            if (result == null){
                tail = result = head;
            } else {
                tail.next = head;
                tail = tail.next;
            }

        }
        return result;
    }
}