1. 程式人生 > >leetcode 23. 合併K個排序連結串列

leetcode 23. 合併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;
    }
};