LeetCode 23.Merge k Sorted Lists (合并K个排序链表)

最新推荐文章于 2021-12-30 01:40:48 发布

amoscykl

最新推荐文章于 2021-12-30 01:40:48 发布

阅读量226

点赞数 1

分类专栏： Leetcode Leetcode (C++解题报告) 文章标签： LeetCode C Merge k Sorted Lists

本文链接：https://blog.csdn.net/amoscykl/article/details/82763061

版权

Leetcode 同时被 2 个专栏收录

144 篇文章 0 订阅

订阅专栏

Leetcode (C++解题报告)

142 篇文章 4 订阅

订阅专栏

题目描述：

合并 k 个排序链表，返回合并后的排序链表。请分析和描述算法的复杂度。

示例：

输入:
[
  1->4->5,
  1->3->4,
  2->6
]
输出: 1->1->2->3->4->4->5->6

Accepted C++ Solution:

使用优先级队列

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */

struct cmp {
    bool operator() (ListNode *a, ListNode *b) {
        return a->val > b->val;        //小的优先级更高
    }
};

class Solution {
public:
    ListNode* mergeKLists(vector<ListNode*>& lists) {
        priority_queue<ListNode*, vector<ListNode*>, cmp> q;    //基于向量的优先级队列
        ListNode* pre = new ListNode(-1);
        ListNode* tmp = pre;
        for (int i = 0; i < lists.size(); i++)
        {
            if (lists[i])
                q.push(lists[i]);       //将所有节点放入优先级队列
        }
        
        while(!q.empty())
        {
            ListNode* t = q.top();
            q.pop();                    
            if (t->next)
                q.push(t->next);        //只需要链表最小的元素，把后续节点重新推回队列
            tmp->next = t;
            tmp = tmp->next;        //tmp = t
        }
        tmp = pre->next;
        delete pre;
        return tmp;
    }
};