LeetCode--Merge k Sorted Lists

Merge k sorted linked lists and return it as one sorted list. Analyze and describe its complexity.

/**
 * 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) 
    {
        int n = lists.size();
        if(n == 0 )
            return NULL;
        if(n == 1)
            return lists[0];
        while(n!=1)
        {
            vector<ListNode*> list_temp;
            for(int i=0; i<n; i=i+2)
            {
                int end = i+1;
                if(i+1>=n)
                    end = n-1;
                if(end == i)
                {
                    list_temp.push_back(lists[i]);
                    break;
                }
                ListNode* temp = mergeTwoLists(lists[i],lists[end]);
                list_temp.push_back(temp);
            }
            lists = list_temp;
            n = lists.size();
        }
        return lists[0];
    }
    ListNode *mergeTwoLists(ListNode *l1, ListNode *l2) 
    {
        if(l1 == NULL)
            return l2;
        if(l2 == NULL)
            return l1;
        ListNode* head;
        ListNode* temp;
        if(l1->val < l2->val)
        {
            head = l1;
            temp = l2;
        }
        else
        {
            head = l2;
            temp = l1;
        }
        ListNode* temp1 = head->next;
        ListNode* now = head;
        while(temp1!=NULL && temp!=NULL)
        {
            if(temp1->val > temp->val)
            {
                now->next = temp;
                now = now->next;
                temp = temp->next;
            }
            else
            {
                now->next = temp1;
                now = now->next;
                temp1 = temp1->next;
            }
        }
        if(temp1 == NULL)
            now->next = temp;
        if(temp == NULL)
            now->next = temp1;
        return head;
    }
};


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值