LeetCode|C++|23. Merge k Sorted Lists

最新推荐文章于 2024-04-03 21:24:25 发布

原创最新推荐文章于 2024-04-03 21:24:25 发布 · 308 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #c++

leetcode 同时被 2 个专栏收录

24 篇文章

订阅专栏

c++

18 篇文章

订阅专栏

本文详细解析了链表合并算法的实现，通过递归调用mergeTwoLists函数，将多个链表两两归并，最终合成一个有序链表。文章提供了完整的代码示例，展示了如何处理链表节点的比较、插入及剩余链表的连接。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

基本思路：两两归并，递归调用mergeTwoLists函数，最终合成一个链表

struct ListNode {
	int val;
	ListNode *next;
	ListNode(int x) : val(x), next(NULL) {}
};

class Solution {
public:
	ListNode* mergeTwoLists(ListNode* l1, ListNode* l2) {
		ListNode *newList = new ListNode(0);
		ListNode *p1 = l1, *p2 = l2, *m = newList;
		while (p1 != NULL&&p2 != NULL){
			if (p1->val <= p2->val){
				ListNode *newnode = new ListNode(p1->val);
				m->next = newnode;
				m = m->next;
				p1 = p1->next;
			}
			else if (p1->val > p2->val){
				ListNode *newnode = new ListNode(p2->val);
				m->next = newnode;
				m = m->next;
				p2 = p2->next;
			}
		}
		if (p1 == NULL&&p2 != NULL)
			m->next = p2;
		else if (p2 == NULL&&p1 != NULL)
			m->next = p1;
		return newList->next;
	}

	ListNode* mergeKLists(vector<ListNode*>& lists) {
		int size = lists.size();
		if (size == 0)
			return NULL;
		else if (size == 1)
			return lists[0];
		else if (size == 2)
			return mergeTwoLists(lists[0], lists[1]);
		else{
			int mid = size / 2;
			vector<ListNode*> lists1, lists2;
			int i = 0;
			while (i < mid)
				lists1.push_back(lists[i++]);
			while (i<size)
				lists2.push_back(lists[i++]);
			ListNode *l1 = mergeKLists(lists1);
			ListNode *l2 = mergeKLists(lists2);
			return mergeTwoLists(l1, l2);

		}
	}
};