23. Merge k Sorted Lists***

原创于 2020-03-14 14:59:21 发布 · 291 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#leetcode #算法

LeetCode 专栏收录该内容

416 篇文章

订阅专栏

本文介绍如何解决LeetCode上的第23题：合并K个已排序的链表并返回一个排序后的链表。通过使用最小堆的方法，详细解释了算法的实现过程和复杂度分析。

23. Merge k Sorted Lists***

https://leetcode.com/problems/merge-k-sorted-lists/

题目描述

Merge k sorted linked lists and return it as one sorted list. Analyze and describe its complexity.

Example:

Input:
[
  1->4->5,
  1->3->4,
  2->6
]
Output: 1->1->2->3->4->4->5->6

C++ 实现 1

使用最小堆处理. 注意输入可能是 [nullptr] 的情况, 因此在下面代码中, 加入了 if (l) 的判断.

class Solution {
private:
    struct Comp {
        bool operator()(ListNode *p1, ListNode *p2) {
            return p1->val > p2->val;
        }
    };
public:
    ListNode* mergeKLists(vector<ListNode*>& lists) {
        priority_queue<ListNode*, vector<ListNode*>, Comp> q;
        ListNode *dummy = new ListNode(0);
        auto ptr = dummy;
        for (auto &l : lists)
            if (l)
                q.push(l);

        while (!q.empty()) {
            auto l = q.top();
            q.pop();
            ptr->next = l;
            ptr = ptr->next;
            if (l->next) q.push(l->next);
        }
        return dummy->next;
    }
};