Merge k Sorted Lists

最新推荐文章于 2021-12-08 15:59:42 发布

原创最新推荐文章于 2021-12-08 15:59:42 发布 · 466 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#LeetCode #合并多个排序链表

LeetCode 专栏收录该内容

36 篇文章

订阅专栏

本文介绍了一种高效的算法来合并多个已排序的链表。该算法基于归并排序的思想，通过递归地将链表分成更小的部分，并最终合并它们以得到一个单一的排序链表。直接逐条合并会导致效率低下，因此文中提出使用二分法进行优化。

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

Merge k Sorted Lists

Merge k sorted linked lists and return it as one sorted list. Analyze and describe its complexity.

解题思路：

目的要合并K个sorted Lists，程序建立在合并两个sorted lists。之后采用二分法对K个list进行合并。整个算法过程类似归并排序。

另外：如果直接逐条合并链表，会超时！！

Code：

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
ListNode *mergeList(ListNode *l1,ListNode *l2)
{
	if(l1==NULL)
		return l2;
	if(l2==NULL)
		return l1;

	ListNode *head=NULL;
	ListNode *cur=NULL;
	ListNode *p=l1;
	ListNode *q=l2;

	while(p && q)
	{
		ListNode *node;
		if(p->val <q->val)
		{
			node=p;
			p=p->next;
		}
		else
		{
			node=q;
			q=q->next;
		}
		if(head==NULL)
			head=cur=node;
		else
		{
			cur->next=node;
			node->next=NULL;
			cur=node;
		}
	}
	if(p)
		cur->next=p;
	else if(q)
		cur->next=q;
	return head;
}

ListNode* helper(vector<ListNode *> &lists,int l,int r)
{
	if(l<r)
	{
		int m=(l+r)/2;
		return mergeList(helper(lists,l,m),helper(lists,m+1,r));
	}
	return lists.at(l);
}

ListNode *mergeKLists(vector<ListNode *> &lists)
{
	if(lists.size()==0)
		return NULL;
	return helper(lists,0,lists.size()-1);
}
};