#每日一题# 23. 合并K个排序链表 - 20191017

题目链接

合并K个排序链表

题目大意

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

示例:

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

解题思路

有暴力法，有按顺序两两归并的方法，有优先队列法，这里讲一下分治的两两归并的方法。

当每个链表中只含一个元素时便是归并排序。本题可以参考归并排序的思想：归并排序采用了分治的方法。为了使整个序列有序，将这个序列分成 2个子序列，先使这 2 个子序列有序然后再将这 2 个子序列合并成一个序列，为了得到有序的子序列又可以对子序列使用归并排序，这样一直递归下去直到子序列中仅含一个元素时，子序列自然有序，然后再从只含一个元素的子序列开始归并，直到生成最终的有序序列。归并排序时间复杂度是 。归并排序是稳定排序。

本题的 K 路归并可以首先将链表数组分为两个子数组，如果每个子数组已经合并成了一个链表，那么问题就转化为合并两个链表，而合并子数组为一个链表与原问题得性质是一样的，因此可以用递归解决，子数组中仅含有一个链表的情况便是递归边界。合并两个有序链表是比较容易写出来的。最终的参考代码如下。

class Solution {
public:
    ListNode* merge(vector<ListNode*>& lists, int l, int r) {
        if(l == r)
            return lists[l];
        else {
            ListNode* l1 = merge(lists, l, (l + r) / 2);
            ListNode* l2 = merge(lists, (l + r) / 2 + 1, r);
            ListNode* head = new ListNode(0);
            ListNode* p = head;
            while(l1 && l2) {
                if(l1->val < l2->val) { 
                    p->next = l1; 
                    l1 = l1->next; 
                }
                else { 
                    p->next = l2; 
                    l2 = l2->next; 
                }
                p = p->next;
            }
            if(l1) p->next = l1;
            if(l2) p->next = l2;
            return head->next;
        }
    }
    ListNode* mergeKLists(vector<ListNode*>& lists) {
        if(lists.empty())
            return NULL;
        else
            return merge(lists, 0, lists.size() - 1);
    }
};

解题感悟

其实应该算比较简单的题目。