首頁
編程語言
C語言|JAVA編程
Python編程
網頁編程
ASP編程|PHP編程
JSP編程
數據庫
MYSQL數據庫|SqlServer數據庫
Oracle數據庫|DB2數據庫
您现在的位置: 程式碼之家 >> 編程語言 >  >> C語言 >> C++ >> 關於C++

C++ priority_queue 輸出最小值(min-heap)

     編輯:關於C++

預設行為

std::priority_queue<T> 內部為 max-heap,top() 回傳最大元素。

改為最小堆

#include <queue>
#include <vector>
#include <functional> // std::greater

// 最小堆:最小元素在 top
priority_queue<int, vector<int>, greater<int>> minHeap;

完整範例

/* main.cpp
 * Created on: 2014.7.20
 */
#include <iostream>
#include <queue>
#include <vector>
#include <functional>
using namespace std;

int main() {
    priority_queue<int, vector<int>, greater<int>> pq;
    for (int x : {5, 2, 8, 1, 9}) pq.push(x);
    while (!pq.empty()) {
        cout << pq.top() << ' ';
        pq.pop();
    }
    cout << endl; // 輸出 1 2 5 8 9
    return 0;
}

自訂比較

若元素為自訂型別,提供 operator> 或自訂仿函式 struct Cmp { bool operator()(const T&a, const T&b) { return a.priority > b.priority; } }。

  1. 上一篇文章:
  2. 下一篇文章:
Copyright © 程式碼之家 All Rights Reserved