Leetcode 23 Merge k Sorted Lists

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
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值