温馨提示×

c++优先队列怎么自定义

c++
小亿
180
2024-02-02 10:05:30
栏目: 编程语言

在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 

通过定义自己的比较函数或类,就可以实现自定义的优先队列排序规则。

0