题解 | #合并k个已排序的链表#
合并k个已排序的链表
http://www.nowcoder.com/practice/65cfde9e5b9b4cf2b6bafa5f3ef33fa6
sort函数进行排序
首先用vector存放所有链表指针,然后sort函数进行排序,注意自写比较函数,最后连接vector中的指针即可。
C++代码:
class Solution {
public:
ListNode *mergeKLists(vector<ListNode *> &lists) {
if (lists.empty()) {
return NULL;
}
vector<ListNode *> ans;
for (int i = 0; i < (int)lists.size(); i++) {
ListNode *cur = lists[i];
while (cur) {
ans.push_back(cur);
cur = cur->next;
}
}
sort(ans.begin(), ans.end(), cmp);
for (int i = 0; i <= (int)ans.size() - 2; i++) {
ans[i]->next = ans[i + 1];
}
ans[ans.size() - 1]->next = NULL;
return ans[0];
}
static bool cmp(ListNode *a, ListNode *b) {
return a->val < b->val;
}
};
时间复杂度O(nlogn): 两次遍历O(n),排序O(nlogn)
空间复杂度O(n):vector存储数据