Leetcode 23 Merge k Sorted Lists

来源:互联网 发布:加州枪击 知乎 编辑:程序博客网 时间:2024/06/07 16:55

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());    }};







1 0
原创粉丝点击