合并 k 个排序链表,返回合并后的排序链表。请分析和描述算法的复杂度。
这题直接用最小堆,第一歩将每个lists的链表头结点放入堆中,第二歩开始循环 每次从堆中取一个元素并放入输出链表的尾端
class Solution {
public:
ListNode *mergeKLists(vector<ListNode *> &lists) {
priority_queue<ListNode*, vector<ListNode*>, cmp> q;
for (int i = 0; i < lists.size(); ++i) {
if (lists[i]) q.push(lists[i]);
}
ListNode *head = NULL, *pre = NULL, *tmp = NULL;
while (!q.empty()) {
tmp = q.top();
q.pop();
if (!pre) head = tmp;
else pre->next = tmp;
pre = tmp;
if (tmp->next) q.push(tmp->next);
}
return head;
}
};
代码中这样设置头结点是非常好的做法,因为这样可以不用重新new一个,节省空间
不像这种