leetcode 23. 合併K個排序連結串列
阿新 • • 發佈:2019-02-13
題目描述:
合併 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; } };