LeetCode 23.Merge k Sorted Lists (合并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;
    }
};

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值