LeetCode 23.合并K个排序链表 C++实现

LeetCode 23.合并K个排序链表 C++实现_第1张图片

 

思路1:按照21的思路,比较所有链表的值,最小的挂到新链表上,比较复杂,时间也很慢  排除

 

思路2:存到数组中,对数组排序,然后再重新链接  时间复杂度(kN*logkN)

struct ListNode{
    int val;
    ListNode * next;
    ListNode(int x) : val(x) , next(NULL){}
};
//因为是结构体,所以要定义一个比较方式
bool cmp(ListNode* a,ListNode* b){
    return a->val < b->val;
}
class Solution {
public:
    ListNode *mergeKLists(vector &lists) {
        vector vec;

        //遍历链表数组,取出每一条链表,然后遍历  加入数组中;
        for(int i = 0 ; i < lists.size() ; i ++){
            ListNode * cur = lists[i];
            while(cur){
                vec.push_back(cur);
                cur = cur->next;
            }
        }
        // 因为下面几处用到vec的大小,所以用变量先接一下可能会快一点,也可以直接用
        int n = vec.size();
        if(n == 0)
            return NULL;
        
        //排序,传入比较函数cmp
        sort(vec.begin(),vec.end(),cmp);

        //链接  第一个连第二个
        for(int i = 1 ; i < n ; i ++){
            vec[i-1]->next =vec[i];
        }
        //最后一个的next为NULL,不然会乱跑 
        vec[n - 1]->next = NULL;
        return vec[0];
    }
};

思路3:

合并 时间复杂度(kN*logk)

每两个链表合并,合并依旧使用21题的代码,然后再接着合并。直到只有一条链表 返回即可

你可能感兴趣的:(LeetCode)