LeetCode 23.合并K个升序链表

最新推荐文章于 2024-10-17 19:06:39 发布

不能熄火

最新推荐文章于 2024-10-17 19:06:39 发布

阅读量56

点赞数

分类专栏： LeetCode (C++ Version) 文章标签： leetcode 链表数据结构

本文链接：https://blog.csdn.net/m0_56728976/article/details/124833523

版权

LeetCode (C++ Version) 专栏收录该内容

123 篇文章 0 订阅

订阅专栏

该博客介绍了一个利用优先队列（最小堆）高效地合并多个已排序链表的算法。通过自定义比较函数，确保堆顶始终是最小的链表节点，从而实现合并过程。代码中创建了一个`cmp`结构体，用于定义比较规则，并在`mergeKLists`函数中实现了合并操作。

摘要由CSDN通过智能技术生成

主页有其他数据结构内容（持续更新中）

难度：Hard

代码：

用优先队列可以很好地解决此类问题

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode() : val(0), next(nullptr) {}
 *     ListNode(int x) : val(x), next(nullptr) {}
 *     ListNode(int x, ListNode *next) : val(x), next(next) {}
 * };
 */
class Solution {
public:
    struct cmp{
        //  重写仿函数
        bool operator()(ListNode* l1, ListNode* l2){
            return l1->val > l2->val; 
            //  这里用>的含义：如果l1->val>l2->val（返回true），那么就进行下沉操作；返回false就不改变
            //  我们的目标是建立一个头部最小的堆
            //  优先队列的源码使用数组实现，array[0]是堆顶，必须使大的数不断下沉
            //  根据所有链表的头结点的比较结果，可以确保优先队列的队头是最小的
        }
    };

    ListNode* mergeKLists(vector<ListNode*>& lists) {
        priority_queue<ListNode*, vector<ListNode*>, cmp> pq;
        for(ListNode* temp : lists){
            if(temp){
                pq.push(temp);
            }
        }
        //  new一个头节点
        ListNode* dummy = new ListNode();
        ListNode* current = dummy;
        while(!pq.empty()){
            current->next = pq.top();   //  指向当前队列顶部结点，也就是最小元素
            pq.pop();   //  把当前结点出队
            current = current->next;
            if(current->next){
                //  如果当前结点存在后继结点，使之入队，继续完成后续比较
                pq.push(current->next);
                //  每次在插入新结点之后，priority_queue都会按照排序规则自动进行一次排序
            }
        }
        return dummy->next;
    }
};