資料結構
網路
RDBMS
作業系統
Java
MS Excel
iOS
HTML
CSS
Android
Python
C 程式設計
C++
C#
MongoDB
MySQL
Javascript
PHP
物理
化學
生物
數學
英語
經濟學
心理學
社會學
時尚研究
法律研究C++ 中最大堆中的第 K 大元素
在本教程中，我們將編寫一個程式，該程式會從最大堆中找出第 K個最大元素。
我們將使用優先順序佇列來解決這個問題。讓我們瞭解完成此程式所需的步驟。
- 以正確的值初始化最大堆。
- 建立一個優先順序佇列並插入最大堆的根節點。
- 編寫一個迴圈，重複 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>> 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>{ 44, 42, 35, 33, 31, 19, 27, 10, 26, 14 };
cout << findKthGreatestElement(heap, 4) << endl;
return 0;
}輸出
如果執行以上程式碼，那麼你將得到以下結果。
33
結論
如果你對本教程有任何疑問，請在評論部分中提及。
更新於：09-4-2021
1 千餘瀏覽量
廣告