[LeetCode 23]合并K个升序链表

[LeetCode 23]C++实现合并K个升序链表

给你一个链表数组，每个链表都已经按升序排列。

请你将所有链表合并到一个升序链表中，返回合并后的链表。

示例 1：

输入：lists = [[1,4,5],[1,3,4],[2,6]]
输出：[1,1,2,3,4,4,5,6]
解释：链表数组如下：
[
1->4->5,
1->3->4,
2->6
]
将它们合并到一个有序链表中得到。
1->1->2->3->4->4->5->6

来源：力扣（LeetCode 23）
链接：https://leetcode-cn.com/problems/merge-k-sorted-lists

方法1、push进vector中排序

class myCompare{
public:
    bool operator()(ListNode* node1,ListNode* node2){
    return node1->val<node2->val;//仿函数由小到大排序
    }
};
class Solution {
public:
ListNode* mergeKLists(vector<ListNode*>& lists){
	if(lists.size()==0)
		return NULL;
    vector<ListNode*>node_V;
    for(int i=0;i<lists.size();++i){
	    //建立临时变量存储各个链表开始节点
        ListNode* head=lists[i];
        while(head){
            node_V.push_back(head);
            head=head->next;//遍历链表
        }
    }
    if(node_V.empty())
        return NULL;
    sort(node_V.begin(),node_V.end(),myCompare());//利用algorithm中sort排序
    for(int i=1;i<node_V.size();++i){
        node_V[i-1]->next=node_V[i];//链接链表
    }
    node_V[node_V.size()-1]->next=NULL;//尾置空
    return node_V[0];//返回头节点
    }  
};

方法2、分治

class Solution {
public:
ListNode* mergeTwoLists(ListNode* node1,ListNode* node2){//合并两个链表//迭代
    if(!node1&&!node2)
        return NULL;
    ListNode* head=new ListNode(0);
    ListNode* ptr=head;
    while(node1&&node2){
        if(node1->val<node2->val){//升序排列
        ptr->next=node1;
        node1=node1->next;
        }
        else{
           ptr->next=node2;
           node2=node2->next;
        }
        ptr=ptr->next;
    }
    if(!node1)
        ptr->next=node2;
    else
        ptr->next=node1;
 //   head->next=NULL;
    return head->next;
}
ListNode* mergeKLists(vector<ListNode*>& lists){
    if(!lists.size())
        return NULL;
    if(lists.size()==1){
        return lists[0];
    }
    if(lists.size()==2){
        return mergeTwoLists(lists[0],lists[1]);
    }
    int mid=lists.size()/2;
    //创建两个子链表
    vector<ListNode*>subList_1;
    vector<ListNode*>subList_2;
	//遍历开始节点
    for(int i=0;i<mid;++i)
        subList_1.push_back(lists[i]);
    for(int i=mid;i<lists.size();++i)
        subList_2.push_back(lists[i]);
	//进入递归
    ListNode* node1=mergeKLists(subList_1);
    ListNode* node2=mergeKLists(subList_2);
    return mergeTwoLists(node1,node2);
    }  
};

ps :mergeTwoLists
https://leetcode-cn.com/problems/merge-two-sorted-lists/ (LeetCode 21)
方法1、迭代

class Solution {
public:
    ListNode* mergeTwoLists(ListNode* l1, ListNode* l2) {
        if(!l1&&!l2)
            return NULL;
        ListNode* head=new ListNode(0);
        ListNode* ptr=head;
        while(l1&&l2){
            if(l1->val<=l2->val){
                ptr->next=l1;
                l1=l1->next;
            }
            else{
                ptr->next=l2;
                l2=l2->next;
            }
            ptr=ptr->next;
        }
        if(!l1)
            ptr->next=l2;
        else    
            ptr->next=l1;
        return head->next;
    }
};

方法2、递归

class Solution {
public:
    ListNode* mergeTwoLists(ListNode* l1, ListNode* l2) {
        if(l1==NULL)
            return l2;
        if(l2==NULL)
            return l1;
        
        if(l1->val<=l2->val){
            l1->next=mergeTwoLists(l1->next,l2);
            return l1;
        }
        else{
            l2->next=mergeTwoLists(l1,l2->next);
            return l2;
        }
    }
};