k个排序链表的合并

LeetCode 23. Merge k Sorted Lists
已知k个已排序链表头节点指针，将这k个链表合并，合并后仍为有序的，返 回合并后的头节点。

思考与分析

最普通的方法，k个链表按顺序合并k-1次，暴力的进行合并。
设有k个链表，平均每个链表有n个节点，时间复杂度: (n+n)+(2n+n)+((k-1)n+n) = (1+2+...+k-1)n +(k-1)n =(1+2+...+k)n-n=(k^2+k-1)/2 * n = O(k^2*n)

是否与更好的方法?

方法1

将kn个节点放到vector中，再将vector排序，再将节点顺序相连。 设有k个链表，平均每个链表有n个节点，时间复杂度:
kNlogkN + kN = O(kN*logkN) (比如k=100, n = 10000) logkN = 20， k = 100

#include
#include
bool cop(const ListNode *a, const ListNode *b){
    return a->val < b-> val;
}
int main(){
    ListNode a(3);
    ListNode b(2);
    ListNode c(5);
    ListNode d(0);
    std::vector node_vec;
    node_vec.push_back(&a);
    node_vec.push_back(&b);
    node_vec.push_back(&c);
    node_vec.push_back(&d);
    std::sort(node_vec.begin(), node_vec.end(),cmp);
    for(int i = 0; i < node_vec.size(); i++){
        printf("%d\n",node_vec[i]->val);
    }
    return 0;
}

#include
#include
bool cmp(const ListNode *a, const ListNode *b){
    return a->val < b->val;
}
class Solution{
public:
    ListNode * mergeKList(std::vector& lists){
    std::vectornode_vec;
        for(int i = 0; i < lists.size();i++){
            ListNode *head = list[i];
            while(head){
                node_vec.push_back(head);
                head = head->next;
        }
        if( node_vec.size() ==0){
           return NULL;
        }
        std::sort(node_vec.begin(),node_vec.end(),cmp);
        for(int i =1; i < node_vec.size(); i++){
            node_vec[i-1]->next = node_vec[i];
        }
        node_vec[node_vec.size()-1]->next = NULL;
        return node_vec[0];

    }
}
};

方法二 分制归并

对k个链表使用分制的思想，两两进行合并。 设有k个链表，平均每个链表有n个节点，时间复杂度: 第1轮，进行k/2次，每次处理2n个数字;第2轮，进行k/4次，每次处理4n个数字;...;
最后一次，进行k/(2^{logk)次，每次处理2}logkN个值。 2Nk/2 + 4N * k/4 + 8N * k/8 +...+2^logk * N * k/(2^logk) =Nk + Nk +...+Nk = O(kNlogk)

ListNode *mergeKList(std::vector &lists){
    if lists.size() == 0){
        return NULL;
    }
    if(lists.size() == 1){
        return list[0];//如果只有一个list，直接返回头节点
    }
    if(lists.size() == 2){
        return mergeTwoLists(lists[0],lists[1]);
    }
    int mid = lists.size() / 2;
    std::vector sub1_lists;
    std::vector sub2_lists;
    for(int i= 0; i < mid ; i++){
        sub1_lists.push_back(lists[i]);
    }
    for(int i = mid; i< lists.size();i++)
{
        sub2_lists.push_back(lists[i]);
    }
    ListNode * l1 = mergeKLists(sub1_lists);
    ListNode *l2= mergeKLists(sub2_lists);
    return mergeTwoList(l1,l2);
}