您的位置:首页 > 其它

Leetcode 23 Merge k Sorted Lists

2016-09-01 20:26 447 查看
Merge k sorted
linked lists and return it as one sorted list. Analyze and describe its complexity.

标为hard难度的一道题,有点interesting,

题意为把K条有序的链表归并为一条有序的链表。

开始想也没想直接暴力n*k的复杂度,果不其然T了,数据出现了n=10000,k=10000的情况,不然怎么对得起hard难度呢?

下面是T掉的代码,

/**
* 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) {
ListNode* result=new ListNode(0);
ListNode* p=result;
while(lists.size()>0)
{
int pos=-1,minn=INT_MAX;
for(int i=0;i<lists.size();i++)
{
if(lists[i]==NULL)
{
lists.erase(lists.begin()+i);
i--;                          //删掉就不存在了,不能忘记下标减一
}
else if(lists[i]->val<minn)
{
minn=lists[i]->val;
pos=i;

}
}
if(pos==-1)
return result->next;
p->next=lists[pos];
p=p->next;
lists[pos]=lists[pos]->next;
}
return result->next;
}
};
然后我开始思考,之前做过两个链表合并的题目

http://blog.csdn.net/accepthjp/article/details/52372112

合并两个链表的复杂度为O(n+m),如果用在这题中,虽然合并次数将很多,但到后来会形成长链和短链的合并,

长链的最后一部分往往是直接接入的,因而实际的平均复杂度并不会很高,但具体是多少,也不太好分析。

这种想法在一般的情况下,应当是优于第一种做法的,

可是后来我想到,如果K个链表都只有一个元素,且随循环合并的方向递增,那么算法又将退化为第一种方法,n=10000,k=10000,随循环合并的方向递增的情况还是会超时。

算法会退化,其实很大程度上还是由于没有改变逐个合并的思路,如果采用分治的方法合并两个链表呢?

很显然,效果会好很多,n*k的复杂度将变为nlogK的复杂度。

/**
* Definition for singly-linked list.
* struct ListNode {
*     int val;
*     ListNode *next;
*     ListNode(int x) : val(x), next(NULL) {}
* };
*/
class Solution {
public:
ListNode* mergeTwoLists(ListNode* l1, ListNode* l2)
{
ListNode* p=new ListNode(0);
ListNode* result=p;
while(l1 && l2)
{
ListNode* t=new ListNode(0);
p->next=t;
p=p->next;
if(l1->val<l2->val)
{
p->val=l1->val;
l1=l1->next;
}
else
{
p->val=l2->val;
l2=l2->next;
}

}
p->next=l1?l1:l2;
return result->next;
}
ListNode* mergeKLists(vector<ListNode*>& lists)
{
while(lists.size()>1)
{
lists.push_back(mergeTwoLists(lists[0],lists[1]));
lists.erase(lists.begin());
lists.erase(lists.begin());
}
if(lists.size()==0)
return NULL;
return *(lists.begin());
}
};
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: