[合并k个升序链表](力扣)

题意

给你一个链表数组,每个链表都已经按升序排列,请你将所有链表合并到一个升序链表中,返回合并后的链表。

样例

输入:lists = [[1, 3, 5], [1, 3, 4], [2, 6]]
输出:[1, 21, 2, 3, 4, 4, 5, 6]
解释:链表数组如下:
[
1->4->5,
1->3->4,
2->6
]
将他们合并到一个有序链表中得到:
1->1->2->3->4->4->5->6

思路

其实给的解释就已经告诉我们应该怎么做了—>分治
加入一共有5个链表,先将1-2、3-4两两合并,得到的结果是(12)、(34)、5三个链表,在将前两个进行合并,得到(1234)、5两个链表,在进行最后的合并。
时间复杂度:O(nlog)

class Solution
{
public:
	ListNode* mergerLists(ListNode* a, ListNode* b)
	{
		if ((!a) || (!b)) retrurn a ? a : b;
		ListNode head, *tail = &head, *pa = a, *pb = b;
		while (pa && pb)
		{
			if (pa->val < pb->val) tail->next = pa, pa = pa->next;
			else tail->next = pb, pb = pb->next;
			tail = tail->next;
		}
		tail->next = (pa ? pa : pb);
		return head.next;
	}
	
	ListNOde* merge(vector<ListNode*> &lists, int l, int r)
	{
		if (l == r) return lsits[l];
		if (l > r) return nullptr;
		
		int mid = (l + r) >> 1;
		return mergerLists(merge(lists, l, mid), merge(lists, mid + 1, r));
	}

	ListNode* mergeKList(vector<ListNode*>& lists)
	{
		return merge(lists, 0, lists.size() - 1);
	}
};

力扣上的方法三:使用优先队列合并:
时间复杂度是一个常数
首先在一个优先队列中,默认是按照从小到大排序的,默认的优先队列是一个大顶堆,队头元素最大。
那么在这里,我们将每个链表的首元素放进优先队列中,放进去之后,每个元素的大小我们让它从小到大进行排序,但是这里我们虽然有了从小到达排序的值,但是对于每个链表中的其他元素该如何去放位置呢?所以这里我们使用一个结构体来存储两个变量:1.值(val), 2.该节点的指针(ptr)。

struct Status
{
	int val;
	ListNode *ptr;
	// 这里我们要使优先队列中的值是从小到大的顺序进行排序,默认是大根堆。
	bool operator < (const Status& rhs) const { return val > rhs.val; }
};

那么我们在遍历优先队列的时候,只有当前值它的后面还有值,那么我们将这个值放入优先队列里面,队列里面会自动将这些值进行排序。

class Soulution
{
public:
	struct Status
	{
		int val;
		ListNode* ptr;
		bool operator < (const Status & rhs) const { return val > ths.val; }	
	};
	
	priority_queue<Status> q;
	ListNode* mergeKLists(vector<ListNode>& lists)
	{
		for (auto node : lists) if (node) q.push({node->val, node});
		ListNode head, *tail = &head;
		while (!q.empty())
		{
			auto f = q.top(); q.pop();
			tail->next = f.ptr;
			tail = tail->next;
			if (f.ptr->next) q.push({f.ptr->val, f.ptr->next});
		}
	};
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值