LeetCode 23. 合并K个升序链表 题解 C/C++

主要思路

前面 #21合并两个有序链表，在这题的基础上顺序合并k个升序链表，定义ans维护结果，不断两两合并即可.

//前面做过合并两个升序链表，这里通过循环，定义ans维护结果，不断两两合并即可
class Solution {
public:
	//合并两个升序链表
	ListNode *mergeTwoLists(ListNode *l1,ListNode *l2) {
		if(!l1||!l2)return l1 ? l1 : l2;
		ListNode fake,*la = l1,*lb = l2;
		ListNode *cur = &fake;
		while(la&&lb) {
			if(la->val < lb->val) {
				cur->next = la;
				la = la->next;
			}
			else {
				cur->next = lb;
				lb = lb->next;
			}
			cur = cur->next;
		}
		cur->next = la ? la : lb;
		return fake.next;
	}

	ListNode *mergeKLists(vector<ListNode*>& lists) {
		/*
		ListNode *ans,*l1,*l2;
		while(lists.size()>1) {
			l1 = lists.back();
			lists.pop_back();
			l2 = lists.back();
			lists.pop_back();
			ans = mergeTwoLists(l1,l2);
			lists.insert(lists.begin(),ans);
			// lists.insert(lists.end(),ans);
			// lists.push_back(ans);
		}
		return lists.size()==1?lists[0]:nullptr;
		*/


		/*不知道相比上面是不是有点慢*/
		ListNode *ans = nullptr;
		for(size_t i = 0;i<lists.size();++i) {//不停的两两合并
			ans = mergeTwoLists(ans,lists[i]);
		}
		return ans;
	}
};