合并 k 个排序链表,返回合并后的排序链表。请分析和描述算法的复杂度。
示例:
输入: [ 1->4->5, 1->3->4, 2->6 ] 输出: 1->1->2->3->4->4->5->6
class Solution {
public:
ListNode* mergeKLists(vector<ListNode*>& lists) {
map<int,ListNode*> count_map;//保存某个值的所有节点构成的单值链表
map<int,ListNode*> head_map;//保存头结点
int size=lists.size();
map<int,ListNode*>::iterator it;
for(int i=0;i<size;i++)
{
ListNode* head=lists[i];
while(head)
{
it=count_map.find(head->val);
if(it!=count_map.end())
{
it->second->next=head;
it->second=it->second->next;
}
else
{//记录头节点
head_map[head->val]=head;
count_map[head->val]=head;
}
head=head->next;
}
}
it=count_map.begin();
map<int,ListNode*>::iterator it_head=head_map.begin();
for(it_head++;it_head!=head_map.end();it_head++,it++)
{//链表拼接
it->next=it_head;
}
return head_map.begin()->second;
}
};
执行用时 : 36 ms, 在Merge k Sorted Lists的C++提交中击败了97.80% 的用户
内存消耗 : 11.5 MB, 在Merge k Sorted Lists的C++提交中击败了73.03% 的用户
1.时间复杂度,数据遍历时,第一层为数组大小n,第二层为最大链表长度m,数据查找负责度为log(数据规模),此处数据规模当链表中所有值都不相同时,数据大小最大为m*n,最小为0 故数据遍历的时间负责度为m*n(log(m*n/2));数据复制时,map大小最大为m*n
故总体时间负责度为m*n(log(m*n/2))