Leetcode Merge k sorted lists

题目

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

题目分析

将k个已经排好序的链表合并成一个sorted的链表。可以采用归并排序的思想：
1. 读出k个链表的链表头，并比较它们的大小
2. 取出最小的那个链表头，并将它连接到新链表的尾部
3. 重复步骤1-2，直至所有链表都为空

源代码

/**
 * 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) {
        vector<ListNode*> headers;
        //initialize
        ListNode* head = NULL, * cur = NULL;
        for (int i = 0; i < lists.size(); i++) 
            if (lists[i] != NULL)
                headers.push_back(lists[i]);
        while (!headers.empty()) {
            ListNode* minNode = new ListNode(0x7FFFFFFF);
            int idx = 0;
            for (int i = 0; i < headers.size(); i++) {
                if (minNode->val > headers[i]->val) {
                    minNode = headers[i];
                    idx = i;
                }
            }
            headers[idx] = headers[idx]->next;
            if (headers[idx] == NULL)
                headers.erase(headers.begin() + idx);
            if (head == NULL) 
                head = minNode;
            else 
                cur->next = minNode;
            cur = minNode;
        }

        return head;
    }
};

改进方案

按照自己的理解做完题目后，去查看discuss区域的solution，列举出几种有用的改进方案:

1. 获取所有的链表头后，需要对它们进行排序，这个排序，我是直接用比较排序，且每次插入完后，其实可以不用再次比较，这里的比较次数很多，算出来的排序这一部分的时间复杂度就会O(sum(Ni))，当总共的元素很多时，就稍显累赘。可以改成：

   • 堆排序，algorithms中可以直接调用make_heap，或者是优先队列（使用堆排序实现）
   • 插值排序

2. 可以参考归并排序的思想，分治法中的merge，因此，这个merge可以用递归的方法，将其分治，也是更容易懂。