Merge k Sorted Lists

最新推荐文章于 2022-07-14 22:28:48 发布

OracleDesire

最新推荐文章于 2022-07-14 22:28:48 发布

阅读量435

点赞数

CC 4.0 BY-SA版权

分类专栏： LeetCode

本文链接：https://blog.youkuaiyun.com/u013086906/article/details/17652567

LeetCode 专栏收录该内容

111 篇文章

订阅专栏

本文介绍了一种使用最小堆实现的K路归并排序方法，通过优先队列维护多个链表头部元素，实现高效合并。文章提供了完整的C++代码示例。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

传说中的K路归并排序问题，显然我们应该用一个最小堆来维护K个链表头里的最小值。

因此这个题的关键就变成了，如何使用STL里的优先队列。

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
struct NodeCmp {
    bool operator() (const ListNode *a, const ListNode *b) {
        return a->val > b->val;
    }
};
class Solution {
public:
    ListNode *mergeKLists(vector<ListNode *> &lists) {
        if (lists.size() == 0)
            return NULL;
        ListNode *fake = new ListNode(-1);
        ListNode *pre = fake;
        priority_queue<ListNode *, vector<ListNode *>, NodeCmp> pq;
        for (int i = 0; i < lists.size(); ++i)
            if (lists[i] != NULL)
                pq.push(lists[i]);
        while (!pq.empty()) {
            ListNode *t = pq.top();
            pq.pop();
            if (t->next != NULL)
                pq.push(t->next);
            pre->next = t;
            pre = t;
            t->next = NULL;
        }
        return fake->next;
    }
    
};

http://oj.leetcode.com/problems/merge-k-sorted-lists/