LeetCode23 Merge k Sorted Lists

题意：

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

分析：

方法1：

利用做过的merge 2 sorted list，将头两个归并，结果再与下一个归并，以此类推，归并完所有。

时间复杂度分析是个小问题， merge 2 sorted list的复杂度是O(n)，本以为结果就是O(n * k)。

但是仔细考虑，随着归并不断进行，其中一个链表的长度不断增加，直至O(n * k)，所以复杂度应该是O(n * k^2) 

代码：

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
private:
     ListNode* mergeTwoLists(ListNode* l1, ListNode* l2) {
        ListNode dummy(0);
        ListNode* head = &dummy;
        while (l1 != nullptr && l2 != nullptr) {
            if (l1 -> val < l2 -> val) {
                head -> next = l1;
                l1 = l1 -> next;
            }
            else {
                head -> next = l2;
                l2 = l2 -> next;
            }
            head = head -> next;
        }
        if (l1 != nullptr) {
            head -> next = l1;
        }
        if (l2 != nullptr) {
            head -> next = l2;
        }
        return dummy.next;
    }
public:
    ListNode* mergeKLists(vector<ListNode*>& lists) {
        if (lists.size() == 0) {
            return nullptr;
        }
        ListNode dummy(0);
        ListNode* head = &dummy;
        head -> next = lists[0];
        for (int i = 1; i < lists.size(); ++i) {
            head -> next = mergeTwoLists(head->next, lists[i]);
        }
        return dummy.next;
        
    }
};

方法2：

利用最小堆。将所有链表的第一个节点加入堆，然后每次取出值最小的节点，并将该节点的next节点加入堆中。堆为空后所以节点处理完，归并后链表。

堆的大小为k，所以插入删除取节点复杂度均为O(logk)，共对O(nk)个节点进行操作，所以时间复杂度为O(nklogk)。

注：

本题除了算法思想本身，还有对于实现过程还是有需要总结的。

1） STL的priority_queue默认的是最大堆，需要重载比较函数。　

　　首先一般来讲priority有三个参数，即

template < class T, class Container = vector<T>,
           class Compare = less<typename Container::value_type> > class priority_queue;

可以看出，2、3有缺省值，即表示用vector实现priority_queue，实现的是最大堆。所以可以不写这两个参数。

但是如果要修改比较方式，即第三个参数的话，第二个参数也要给出（以前没用过，真是查文档才知道）。

对于内置类型，如int，可以使用stl的functor如 greater<int>直接修改堆的大小属性，但是自定义节点，如本例的ListNode，需要自己重载。

2）对于传入比较函数，一种方法是重载类的operator <，但是leetcode中ListNode的定义是不能修改的。

　　还有就是写一个cmp函数，值得注意的是。funtor作为一个类需要重载其operator()，即对于本题，写为

    struct cmp{
      bool operator() (ListNode* l1, ListNode* l2) {  
          return l1 -> val > l2 -> val;
      }  
    };  

另外，priority_queue的第三个参数不能只传函数，必须封装成个functor（类），开始写的时候传函数报错了。

总之这题收货还是不少，不论从算法角度还是语言角度。

 

代码：

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
private:
    struct cmp{
      bool operator() (ListNode* l1, ListNode* l2) {  //why?!
          return l1 -> val > l2 -> val;
      }  
    };  
public:
    ListNode* mergeKLists(vector<ListNode*>& lists) {
        ListNode dummy(0);
        ListNode* head = &dummy;
        priority_queue<ListNode*, vector<ListNode*>, cmp> que;
        for (int i = 0; i < lists.size(); ++i) {
            if (lists[i] != nullptr) {
                que.push(lists[i]);
            }
        }
        while (!que.empty()) {
            ListNode* temp = que.top();
            head -> next = temp;
            head = head -> next;
            que.pop();
            if (temp -> next != nullptr) {
                que.push(temp->next);
            }
        }
        return dummy.next;
    }    
};