You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
67 lines
1.7 KiB
67 lines
1.7 KiB
/**
|
|
* File: heap.cpp
|
|
* Created Time: 2023-01-19
|
|
* Author: LoneRanger(836253168@qq.com)
|
|
*/
|
|
|
|
#include "../utils/common.hpp"
|
|
|
|
void testPush(priority_queue<int> &heap, int val) {
|
|
heap.push(val); // 元素入堆積
|
|
cout << "\n元素 " << val << " 入堆積後" << endl;
|
|
printHeap(heap);
|
|
}
|
|
|
|
void testPop(priority_queue<int> &heap) {
|
|
int val = heap.top();
|
|
heap.pop();
|
|
cout << "\n堆積頂元素 " << val << " 出堆積後" << endl;
|
|
printHeap(heap);
|
|
}
|
|
|
|
/* Driver Code */
|
|
int main() {
|
|
/* 初始化堆積 */
|
|
// 初始化小頂堆積
|
|
// priority_queue<int, vector<int>, greater<int>> minHeap;
|
|
// 初始化大頂堆積
|
|
priority_queue<int, vector<int>, less<int>> maxHeap;
|
|
|
|
cout << "\n以下測試樣例為大頂堆積" << endl;
|
|
|
|
/* 元素入堆積 */
|
|
testPush(maxHeap, 1);
|
|
testPush(maxHeap, 3);
|
|
testPush(maxHeap, 2);
|
|
testPush(maxHeap, 5);
|
|
testPush(maxHeap, 4);
|
|
|
|
/* 獲取堆積頂元素 */
|
|
int peek = maxHeap.top();
|
|
cout << "\n堆積頂元素為 " << peek << endl;
|
|
|
|
/* 堆積頂元素出堆積 */
|
|
testPop(maxHeap);
|
|
testPop(maxHeap);
|
|
testPop(maxHeap);
|
|
testPop(maxHeap);
|
|
testPop(maxHeap);
|
|
|
|
/* 獲取堆積大小 */
|
|
int size = maxHeap.size();
|
|
cout << "\n堆積元素數量為 " << size << endl;
|
|
|
|
/* 判斷堆積是否為空 */
|
|
bool isEmpty = maxHeap.empty();
|
|
cout << "\n堆積是否為空 " << isEmpty << endl;
|
|
|
|
/* 輸入串列並建堆積 */
|
|
// 時間複雜度為 O(n) ，而非 O(nlogn)
|
|
vector<int> input{1, 3, 2, 5, 4};
|
|
priority_queue<int, vector<int>, greater<int>> minHeap(input.begin(), input.end());
|
|
cout << "輸入串列並建立小頂堆積後" << endl;
|
|
printHeap(minHeap);
|
|
|
|
return 0;
|
|
}
|