合并k个有序链表

题目描述

这是leetcode上面的一道编程题，题目如下

/*
合并 k 个排序链表，返回合并后的排序链表。请分析和描述算法的复杂度。

示例

输入:
[
  1->4->5,
  1->3->4,
  2->6
]
输出: 1->1->2->3->4->4->5->6

*/

---

解题思路

利用递归分治的思想将k个有序链表的合并问题，分解成多个合并两个有序链表的问题。比如示例中我们将三个链表分成两份，第一个链表单独为一份，第二个和第三个链表为一份，然后将第二个和第三个按照两个有序链表合并的思路进行合并，之后再将得到的结果和第一个链表进行合并，就可以得到一个有序的链表。

---

实现代码

ListNode* mergeKLists(vector<ListNode*>& lists, int begin, int end) {
        if (end < begin) return NULL;
        if (end == begin) return lists[begin];

        int mid = (begin + end) / 2;
        auto p1 = mergeKLists(lists, begin, mid);
        auto p2 = mergeKLists(lists, mid + 1, end);

        ListNode head(0);
        ListNode* cur = &head;
        while (p1 && p2) {
            if (p1->val < p2->val) {
                cur->next = p1;
                p1 = p1->next;
            } else {
                cur->next = p2;
                p2 = p2->next;
            }
            cur = cur->next;
        }

        if (p1) cur->next = p1;
        else if (p2) cur->next = p2;
        return head.next;
    }

    ListNode* mergeKLists(vector<ListNode*>& lists) {
        return mergeKLists(lists, 0, lists.size() - 1);
    }