leetcode 23. 合并K个排序链表

题目描述：

 

---

合并 k 个排序链表，返回合并后的排序链表。请分析和描述算法的复杂度。

示例:

输入:
[
  1->4->5,
  1->3->4,
  2->6
]
输出: 1->1->2->3->4->4->5->6

 

代码：

class Solution {
public:
    ListNode* mergeKLists(vector<ListNode*>& lists) {
        if(lists.size()==0)return NULL;
        if(lists.size()==1)return lists[0];
        ListNode* head=lists[0];
        for(int i=1;i<lists.size();i++){
            head=merge(head,lists[i]);
        }
        return head;
    }
    ListNode* merge(ListNode*head1,ListNode*head2){
        ListNode*head=NULL,*cur=NULL;
        while(head1&&head2){
            if(head1->val>head2->val){
                ListNode*temp=head1;
                head1=head2;
                head2=temp;
            }    
            if(head){
                cur=cur->next=head1;
                head1=head1->next;
            }
            else{
                head=cur=head1;
                head1=head1->next;
            }
        }
        if(head1&&head)cur->next=head1;
        if(head2&&head)cur->next=head2;
        if(head1&&!head)head=head1;//传递进来的head2为空的情况
        if(head2&&!head)head=head2;//传递进来的head1为空的情况
        return head;
    }
};