[各种面试题] 合并k个排序链表

最新推荐文章于 2024-01-08 14:15:07 发布

原创最新推荐文章于 2024-01-08 14:15:07 发布 · 1.7k 阅读

0 ·

CC 4.0 BY-SA版权

各种面试题专栏收录该内容

66 篇文章

订阅专栏

本文介绍了一种使用优先队列实现多个有序链表高效合并的方法。通过自定义比较器，确保每次都能从队列中取出最小值，从而完成链表的正确合并。

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

思路是用一个优先队列，给它自定义一个排序函数即可。

/*链表结点的定义(请不要在代码中定义该类型)
struct ListNode {
  int val;
  ListNode *next;
};
*/
//lists包含k个链表的头结点,返回合并后链表头结点
const int INT_MAX=2147483647;
class compare
{
public:
	bool operator()(ListNode* n1,ListNode* n2)const
	{
		if ( !n1 || !n2)
			return !n1;
		return n1->val>n2->val;
	}
};

ListNode* merge(vector<ListNode*> &lists) {
	if ( lists.empty() )
		return NULL;
	ListNode guard;
	ListNode* tail=&guard;
	priority_queue<ListNode*,vector<ListNode*>,compare> Q;
	for(int i=0;i<lists.size();i++)
		Q.push(lists[i]);
	while(!Q.empty()&&Q.top()!=NULL)
	{
		ListNode* t=Q.top();
		Q.pop();
		tail->next=t;
		tail=tail->next;
		Q.push(t->next);
	}
	return guard.next;
}