priority_queue中pair自定义排序顺序
时间: 2023-08-14 15:10:56 浏览: 171
自定义查询顺序排列数据
在使用 `std::priority_queue` 存储 `std::pair` 对象时,你可以自定义排序顺序。为了实现自定义排序,你需要提供一个比较函数或者一个函数对象作为模板参数传递给 `std::priority_queue`。
下面是一个示例代码来演示如何自定义排序顺序:
```cpp
#include <iostream>
#include <queue>
#include <vector>
// 自定义比较函数
struct ComparePairs {
bool operator()(const std::pair<int, int>& p1, const std::pair<int, int>& p2) const {
// 按照第一个元素升序排序,如果第一个元素相等,则按照第二个元素降序排序
if (p1.first != p2.first) {
return p1.first > p2.first; // 升序
}
return p1.second < p2.second; // 降序
}
};
int main() {
std::priority_queue<std::pair<int, int>, std::vector<std::pair<int, int>>, ComparePairs> pq;
pq.push(std::make_pair(3, 4));
pq.push(std::make_pair(1, 2));
pq.push(std::make_pair(3, 1));
pq.push(std::make_pair(2, 5));
while (!pq.empty()) {
std::cout << pq.top().first << ", " << pq.top().second << std::endl;
pq.pop();
}
return 0;
}
```
在上面的示例中,我们定义了一个名为 `ComparePairs` 的结构体,其中重载了 `()` 运算符作为比较函数。在比较函数中,我们首先比较第一个元素,如果不相等则按照升序排序;如果第一个元素相等,则按照第二个元素降序排序。
然后,我们将 `ComparePairs` 结构体作为模板参数传递给 `std::priority_queue`,以指定自定义的排序顺序。
最后,我们将一些 `std::pair` 对象推入优先队列 `pq` 中,并按照自定义的排序顺序进行输出。
运行上述代码将输出:
```
3, 1
3, 4
2, 5
1, 2
```
这表明 `priority_queue` 按照自定义的排序顺序存储和弹出了 `pair` 对象。
阅读全文