是的,C++的PriorityQueue
允许你自定义比较函数。你可以通过传递一个自定义的比较对象(也称为仿函数)给PriorityQueue
的构造函数来实现这一点。这个比较对象需要定义一个名为operator()
的成员函数,该函数接受两个参数并返回一个布尔值,用于比较这两个参数。
以下是一个使用自定义比较函数的PriorityQueue
示例:
#include <iostream>
#include <queue>
// 自定义比较函数对象
struct Compare {
bool operator()(int a, int b) {
return a > b; // 降序排列
}
};
int main() {
// 使用自定义比较函数创建 PriorityQueue
std::priority_queue<int, std::vector<int>, Compare> pq;
// 向 PriorityQueue 中添加元素
pq.push(5);
pq.push(3);
pq.push(8);
pq.push(1);
// 输出 PriorityQueue 中的元素
while (!pq.empty()) {
std::cout << pq.top() << " ";
pq.pop();
}
return 0;
}
在这个示例中,我们定义了一个名为Compare
的结构体,它重载了operator()
函数以实现降序排列。然后,我们使用这个比较对象创建了一个PriorityQueue
实例。注意,我们传递了一个std::vector<int>
作为底层容器,因为PriorityQueue
默认使用std::vector
作为底层容器。