题目描述
23.合并K个排序链表
合并 k 个排序链表,返回合并后的排序链表。请分析和描述算法的复杂度。
示例
输入:
[
1->4->5,
1->3->4,
2->6
]
输出: 1->1->2->3->4->4->5->6
思路
1.暴力法
- 将链表中所有元素加入到一个数组中
- 对数组进行排序
- 新建链表,将数组中元素放入链表中。
- 时间复杂度 O(nlogn) ,***n***为链表中总结点个数
- 所需空间 O(n) ,新建链表 O(n) ,数组空间 O(n)
代码如下
class Solution {
public:
ListNode* mergeKLists(vector<ListNode*>& lists) {
vector<int> ans;
for(int i = 0; i < lists.size(); i++)
{
while(lists[i] != NULL)
{
ans.push_back(lists[i]->val);
lists[i] = lists[i]->next;
}
}
sort(ans.begin(), ans.end());
ListNode* p = new ListNode(0);
ListNode* q = p;
for(int i = 0; i <ans.size(); i++)
{
p->next = new ListNode(ans[i]);
p = p->next;
}
return q->next;
}
};