数据结构
网络
RDBMS
操作系统
Java
MS Excel
iOS
HTML
CSS
Android
Python
C 编程
C++
C#
MongoDB
MySQL
Javascript
PHP
物理
化学
生物
数学
英语
经济学
心理学
社会学
时尚学
法律学C++ 中 Min 堆中的第 K 小元素
在本教程中，我们将编写一个程序，从 min 堆中找到第 k 小的元素。
我们将使用优先级队列来解决这个问题。我们来看一下完成该程序的步骤。
- 用正确的值初始化 min 堆。
- 创建一个优先级队列，并插入 min 堆的根节点。
- 编写一个循环，迭代 k - 1 次。
- 从队列中弹出最小的元素。
- 将上述节点的左右节点添加到优先级队列中。
- 现在优先级队列中最大的元素是第 k 个最大的元素。
- 返回该元素。
示例
我们来看一下代码。
#include <bits/stdc++.h>
using namespace std;
struct Heap {
vector<int> elemets;
int n;
Heap(int i = 0): n(i) {
elemets = vector<int>(n);
}
};
inline int leftIndex(int i) {
return 2 * i + 1;
}
inline int rightIndex(int i) {
return 2 * i + 2;
}
int findKthGreatestElement(Heap &heap, int k) {
priority_queue<pair<int, int>, vector<pair<int, int>>, greater<pair<int, int>>>queue;
queue.push(make_pair(heap.elemets[0], 0));
for (int i = 0; i < k - 1; ++i) {
int node = queue.top().second;
queue.pop();
int left = leftIndex(node), right = rightIndex(node);
if (left < heap.n) {
queue.push(make_pair(heap.elemets[left], left));
}
if (right < heap.n) {
queue.push(make_pair(heap.elemets[right], right));
}
}
return queue.top().first;
}
int main() {
Heap heap(10);
heap.elemets = vector<int>{ 10, 14, 19, 24, 32, 41, 27, 44, 35, 33 };
cout << findKthGreatestElement(heap, 4) << endl;
return 0;
}输出
如果您运行以上代码，您将获得以下结果。
24
结论
如果您对本教程有任何疑问，请在评论部分提及。
更新时间： 2021 年 4 月 9 日
475 次浏览
广告