Merge k Sorted Lists（合并 K 個(gè)已排序鏈表）

翻譯

合并 K 個(gè)已排序的鏈表，并且將其排序并返回。
分析和描述其復(fù)雜性。

原文

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

代碼

我們采用分治的方法來解決這個(gè)問題，其有 K 個(gè)鏈表，不斷將其劃分（partition），再將其歸并（merge）。

劃分的部分并不難，將其不斷分成兩部分，但是需要注意的是可能出現(xiàn) start 和 end 相等的情況，這時(shí)候就直接 return lists[start] 就可以了。

mid = (start + end) / 2
start -- mid                    （1）
mid + 1 -- end                  （2）

上面的（1）和（2）就不斷的替換更新，不斷作為參數(shù)寫到 partition 函數(shù)內(nèi)。

歸并的部分也不難，因?yàn)樵诖酥拔覀円呀?jīng)遇到過，傳送門：LeetCode 21 Merge Two Sorted Lists 。

所以結(jié)合起來就是下面這樣的：

/**
 * 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) {
        return partition(lists, 0, lists.size() - 1);
    }

    ListNode* partition(vector<ListNode*>& lists, int start, int end) {
        if(start == end) {
            return lists[start];
        }

        if(start < end) {
            int mid = (start + end) / 2;            
            ListNode* l1 = partition(lists, start, mid);
            ListNode* l2 = partition(lists, mid + 1, end);
            return mergeTwoLists(l1, l2);
        }       
        return NULL;
    }

    ListNode* mergeTwoLists(ListNode* l1, ListNode* l2) {
        if(l2 == NULL) return l1;
        if(l1 == NULL) return l2;

        if(l1->val > l2->val) {
            ListNode* temp = l2;
            temp->next = mergeTwoLists(l1, l2->next);
            return temp;
        } else {
            ListNode* temp = l1;
            temp->next = mergeTwoLists(l1->next, l2);
            return temp;
        }
    }
};

繼續(xù)學(xué)習(xí)大神的的解法：

/**
 * 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) {
        int size = lists.size();
        if(size == 0) return NULL;
        if(size == 1) return lists[0];

        int i = 2, j;
        while(i / 2 < size) {
            for(j = 0; j < size; j += i) {
                ListNode* p = lists[j];
                if(j + i / 2 < size) {
                    p = mergeTwoLists(p, lists[j + i / 2]);
                    lists[j] = p;
                }
            }
            i *= 2;
        }
        return lists[0];
    }

    ListNode* mergeTwoLists(ListNode* l1, ListNode* l2) {
        if(l2 == NULL) return l1;
        if(l1 == NULL) return l2;

        if(l1->val > l2->val) {
            ListNode* temp = l2;
            temp->next = mergeTwoLists(l1, l2->next);
            return temp;
        } else {
            ListNode* temp = l1;
            temp->next = mergeTwoLists(l1->next, l2);
            return temp;
        }
    }
};