23. 合并K个排序链表
合并 k 个排序链表,返回合并后的排序链表。请分析和描述算法的复杂度。
Solution1:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29
| bool cmp(const ListNode* a,const ListNode *b){ return a->val < b->val; } class Solution { public: ListNode* mergeKLists(vector<ListNode*>& lists) { std::vector<ListNode*> node_vec; for(int i = 0; i < lists.size(); i++){ ListNode* head = lists[i]; while(head){ node_vec.push_back(head); head = head->next; } } if(node_vec.size() == 0){ return NULL; } std::sort(node_vec.begin(),node_vec.end(),cmp); for(int i = 1;i < node_vec.size();i++){ node_vec[i-1]->next = node_vec[i]; } node_vec[node_vec.size()-1]->next = NULL; return node_vec[0]; } };
|
该方法比较粗暴。将所有的结点存入一个vector中,按结点中的值排序,然后依次将每个结点链接,最后返回。
Solution2:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48
| class Solution { public: ListNode* mergeTwoLists(ListNode* l1, ListNode* l2) { ListNode p0 = ListNode(0); ListNode* p = &p0; ListNode* p1 = p; while(l1 && l2){ if(l1->val < l2->val){ p->next = new ListNode(l1->val); p = p->next; l1 = l1->next; }else{ new ListNode(l2->val); p->next = new ListNode(l2->val); p = p->next; l2 = l2->next; } } p->next = (l1!=NULL) ? l1 :(l2!=NULL) ? l2 : NULL; return p1->next; } ListNode* mergeKLists(vector<ListNode*>& lists) { //两个终止条件。 if(lists.size() == 0){ return NULL; } if(lists.size() == 1){ return lists[0]; } //将lists拆分成两个子lists; int mid = lists.size()/2; std::vector<ListNode*> sub1_lists; std::vector<ListNode*> sub2_lists; for(int i = 0; i < mid; i++){ sub1_lists.push_back(lists[i]); } for(int i = mid; i < lists.size(); i++){ sub2_lists.push_back(lists[i]); } //递归调用 ListNode *l1 = mergeKLists(sub1_lists); ListNode *l2 = mergeKLists(sub2_lists); //返回两个链表合并的结果。 return mergeTwoLists(l1,l2); } };
|
思路:
(归并思想)递归算法。
将k个链表分割为独立的链表,再向上两两合并。
合并两个有序链表的子问题参照:
21. 合并两个排序链表