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& lists) {
        if(lists.size()==0)return NULL;
        if(lists.size()==1)return lists[0];
        ListNode* head=lists[0];
        for(int i=1;ival>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;
    }
};

 

你可能感兴趣的:(leetcode,算法编程)