LeetCode——第二十二天
23. 合并K个排序链表
合并 k 个排序链表,返回合并后的排序链表。请分析和描述算法的复杂度。
示例:
输入:
[
1->4->5,
1->3->4,
2->6
]
输出: 1->1->2->3->4->4->5->6
分析:只会暴力,看了官方分治,还有一个优先队列就比较好玩了。分治我没有去写,这边写下优先队列,其实也挺简单,就是比如例题,因为本身都是排序好的,优先队列会维护自己排序,然后把最小的pop后加入最后链表,被选中的那个就直接next,重复就好了。不过简单的原因也是没有自己写优先队列,自己写一个还是挺麻烦的。
优先队列
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
class Solution {
public:
struct Status {
int val;
ListNode *ptr;
//重置<运算符
bool operator < (const Status &rhs) const {
return val > rhs.val;
}
};
priority_queue <Status> q;//优先队列
ListNode* mergeKLists(vector<ListNode*>& lists) {
for (auto node: lists) {
if (node) q.push({node->val, node});//头结点按顺序加入
}
ListNode head, *tail = &head;
while (!q.empty()) {
auto f = q.top(); q.pop();//弹出最小值,加入tail链表中
tail->next = f.ptr;
tail = tail->next;
if (f.ptr->next) q.push({f.ptr->next->val, f.ptr->next});//当前退出值下一个如果非空则加入优先队列中
}
return head.next;
}
};