leetcode 23. Merge k Sorted Lists

 * 相当于归并排序的归并部分,参考讲解:https://www.youtube.com/watch?v=6bvnZzwiKzs

 k个有序的链表,假设每个链表的长度为n,将这k个链表合并成一个

 有序链表。

 一种方法是,直接拼接这k个链表后再排序,复杂度是O(KN*LOG(KN))

 另一种方法是,每次在k个链表中选出排在第一的最小的元素,如此下去,

 复杂度是O(KN*K)

 还有一种方法是上一种方法的基础上加一个priority_queue,可以将复杂度

 降低到O(KN*LOG(K))

struct ListNode {
    int val;
    ListNode *next;
    ListNode(int x) : val(x), next(NULL) {}
};

class Solution {
public:
    ListNode* mergeKLists(vector<ListNode*>& lists) {
        ListNode *head = NULL, *tail = NULL;;
        priority_queue<ListNode*, vector<ListNode*>, function<bool(const ListNode * leftPtr, ListNode *rightPtr)>> cachQueue([](const ListNode *leftPtr, const ListNode *rightPtr) {return leftPtr->val < rightPtr->val;});
        for (int i = 0; i < lists.size(); i++) {
            if (lists[i] != NULL) {
                cachQueue.push(lists[i]);
                lists[i] = lists[i]->next;
            }
        }
        if (!cachQueue.empty()) {
            head = cachQueue.top();
            cachQueue.pop();
            if (head->next != NULL) {
                cachQueue.push(head->next);
            }
            tail = head;
        }
        while (!cachQueue.empty()) {
            ListNode *newPtr = cachQueue.top();
            cachQueue.pop();
            if (newPtr->next != NULL) {
                cachQueue.push(newPtr->next);
            }
            tail->next = newPtr;
            tail = newPtr;
        }
        return head;
    }
};


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值