【LeetCode算法练习(C++)】Merge k Sorted Lists

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

链接:Merge k Sorted Lists
解法:递归实现归并,利用merge Two Lists,每次合并两个序列。时间O(nlogn)


class Solution {
public:
    ListNode* mergeKLists(vector<ListNode*>& lists) {
        if (lists.size() == 0) return NULL;
        else return VmergeKLists(lists)[0];
    }

    vector<ListNode*> VmergeKLists(vector<ListNode*>& lists) {
        vector<ListNode*> res;
        if (lists.size() == 1) return lists;
        else if (lists.size() == 2) {
            res.push_back(mergeTwoLists(lists[0], lists[1]));
            return res;
        }
        else {
            vector<ListNode*> vl;
            for (int i = 1; i < lists.size(); i += 2) {
                vl.push_back(mergeTwoLists(lists[i - 1], lists[i]));
            }
            if (lists.size() % 2) vl.push_back(lists[lists.size() - 1]);
            return VmergeKLists(vl);
        }
    }

    ListNode* mergeTwoLists(ListNode* l1, ListNode* l2) {
        ListNode *head = new ListNode(0);
        head->next = NULL;
        ListNode *p = head;
        while (l1 || l2) {
            ListNode *tmp = new ListNode(0);
            tmp->next = NULL;
            if (!l1 && l2) {
                tmp->val = l2->val;
                l2 = l2->next;
            } else if (!l2 && l1) {
                tmp->val = l1->val;
                l1 = l1->next;
            } else if (l1->val > l2->val) {
                tmp->val = l2->val;
                l2 = l2->next;
            } else {
                tmp->val = l1->val;
                l1 = l1->next;
            }
            p->next = tmp;
            p = p->next;
        }
        p = head;
        head = head->next;
        p->next = NULL;
        delete(p);
        return head;
    }
};

Runtime: 32 ms

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值