题目描述:合并 k 个排序链表,返回合并后的排序链表。请分析和描述算法的复杂度。
示例:
输入:
[
1->4->5,
1->3->4,
2->6
]
输出: 1->1->2->3->4->4->5->6
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/merge-k-sorted-lists
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路:
1、将所有链表中的值放入vector容器中;
2、使用sort()对vector中的值进行排序;
3、将排序后的vector中的数保存到新的链表中,返回链表head指针
空间复杂度O(N),时间复杂度O(N)
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
class Solution {
public:
ListNode* mergeKLists(vector<ListNode*>& lists) {
if (lists.size() == 0) {
return nullptr;
}
if (lists.size() == 1) {
return lists[0];
}
vector<int> res_list;
// 将所有链表中的值放入vector中
for(int i=0;i<lists.size();i++){
ListNode* now = lists[i];
while(now){
res_list.push_back(now->val);
now = now->next;
}
}
sort(res_list.begin(),res_list.end()); // 排序
// 将vector中的值放入新创建的链表中
ListNode* cur = new ListNode(NULL);
ListNode* head = cur;
for(int i = 0;i < res_list.size();i++)
{
cur->val = res_list[i];
if(i < res_list.size()-1)
cur->next = new ListNode(NULL);
cur = cur->next;
}
return head;
}
};