阅读量:0
在C++中,可以使用std::priority_queue
类来实现优先队列。默认情况下,std::priority_queue
是使用std::less
来进行比较的,即元素的优先级由最大的元素排在队列前面。如果要自定义优先队列的排序规则,可以通过使用自定义的比较函数或者自定义的类来实现。
使用自定义的比较函数:
#include <iostream> #include <queue> struct CustomCompare { bool operator()(int a, int b) { // 自定义排序规则,按照元素的绝对值大小进行排序 return std::abs(a) > std::abs(b); } }; int main() { std::priority_queue<int, std::vector<int>, CustomCompare> pq; pq.push(3); pq.push(-5); pq.push(2); pq.push(-1); while (!pq.empty()) { std::cout << pq.top() << " "; pq.pop(); } return 0; }
输出结果:
-5 3 -1 2
使用自定义的类:
#include <iostream> #include <queue> class MyClass { public: int value; MyClass(int v) : value(v) {} }; struct CustomCompare { bool operator()(const MyClass& a, const MyClass& b) { // 自定义排序规则,按照元素的value进行排序 return a.value > b.value; } }; int main() { std::priority_queue<MyClass, std::vector<MyClass>, CustomCompare> pq; pq.push(MyClass(3)); pq.push(MyClass(-5)); pq.push(MyClass(2)); pq.push(MyClass(-1)); while (!pq.empty()) { std::cout << pq.top().value << " "; pq.pop(); } return 0; }
输出结果:
-5 -1 2 3
通过定义自己的比较函数或类,就可以实现自定义的优先队列排序规则。