Merge k Sorted Lists

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

堆实现:

class Solution {
public:
    struct cmp{
        bool operator()(ListNode* node1, ListNode* node2){
            return node1->val > node2->val;
        }
    };
    ListNode *mergeKLists(vector<ListNode *> &lists) {
        ListNode *node = NULL;
        ListNode *head = NULL;
        priority_queue<ListNode*,vector<ListNode*>,cmp>temp;
        for(int i = 0; i < lists.size(); i++){
            if(lists[i] != NULL){
                temp.push(lists[i]);
            }
        }
        while(!temp.empty()){
            ListNode *p = temp.top();
            temp.pop();
            if(node == NULL){
				head = p;
			}
			else{
				node->next = p;
			}
            if(p->next){
                temp.push(p->next);
            }
            node = p;
        }
        return head;
    }
};

归并实现:

class Solution {
public:
    ListNode *mergeKLists(vector<ListNode *> &lists) {
        if(lists.size() == 0){
            return NULL;
        }
        return merge(lists, 0,  lists.size() - 1);
    }
    ListNode *merge(vector<ListNode *>&lists, int left, int right){
        if(left < right){
            int m = (left + right) / 2;
            return twoMerge(merge(lists,left, m), merge(lists, m + 1, right)); 
        }
        return lists[left];
    }
    ListNode *twoMerge(ListNode *l1, ListNode *l2){
        ListNode *dummy = new ListNode(0);
        ListNode *curr = dummy;
        while(l1 != NULL && l2 != NULL){
            if(l1->val < l2->val){
                curr->next = l1;
                l1 = l1->next;
            }
            else{
                curr->next = l2;
                l2 = l2->next;
            }
            curr = curr->next;
        } 
        if(l1 != NULL){
            curr->next = l1;
        }
        if(l2 != NULL){
            curr->next = l2;
        }
        return dummy->next;
    }
};

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值