优先队列,二叉堆以及topK问题

1.优先队列是按照优先级排序的队列,优先级大(默认)的先出。

2.二叉堆是一颗完全二叉树。
树的任意节点值大于等于其子节点:最大堆。
树的任意节点值小于等于其子节点:最小堆
完全二叉树通常使用数组存储,节点之间对应的下标关系是:某一结点下标i,则父节点下标[(i-1)/ 2], 左节点2 * i+1,右节点2 * i + 2.
可以认为优先队列就是实现大顶堆或小顶堆的数组。
在这里插入图片描述
这是一个大顶堆,所以说优先队列(图中数组)内的元素并不是严格排序的,要想得到排序后的元素还要依次出队,将元素存入新的容器里。

在构建二叉堆时:
在这里插入图片描述
在这里插入图片描述在这里插入图片描述
插入过程(上浮):从数组第0个元素开始不断插入大根堆中。如果该元素大于父节点,就与父节点交换,直到根节点。

删除操作(下沉):将堆顶元素与最后一个元素交换,然后将新的堆顶元素不断下沉直到所在的树满足小顶堆关系。(与左右孩子较大者比较,如小于较大者则交换,否则表示调整结束)

插入和删除操作分别对应优先队列的入队和出队操作。代码见“”排序”中堆排序部分。

3.c++优先队列

priority_queue<int>;
priority_queue<int,vector<int>, cmp>;

优先队列数值大的优先级高,用来实现大顶堆的,这是默认情况;数值小的优先级高的队列用来实现小顶堆。因为对顶元素放在数组第一个位置上。
其中cmp可以有less或者greater代替,使用less的时候是正常大小关系,数值大的认为优先大,因此大根堆;使用greater是相反。priority_queue<int>默认是使用less, 实现的大根堆。

对于自定义类型,可以实现其<操作符.

struct Node {
	int v;
	int u;
	Node(int a, int b) :v(a), u(b) {}

	friend bool operator< (const Node& a, const Node& b);
};

bool operator< (const Node& a, const Node& b)
{
	return a.u > b.u;
}

topK问题:提取数组的前k最大数。
使用优先队列实现小顶堆,优先队列中一直维持着k个数,当元素数大于k时,去掉对顶元素(最小值)。最后队列里保存k个top值。

vector<int> topk(vector<int> &num, int k)
{
	vector<int> res;
	priority_queue<int, vector<int>, greater<int>> qu;
	if (k <= 0) return {};

	for (int i : num)
	{
		qu.push(i);
		if (qu.size() > k) qu.pop();
	}
	while (!qu.empty())
	{
		res.push_back(qu.top());
		qu.pop();
	}
	return res;
}

参考:
https://www.cnblogs.com/wmyskxz/p/9301021.html

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值