温馨提示×

C++ PriorityQueue 能否自定义比较函数

c++
小樊
83
2024-10-14 18:29:08
栏目: 编程语言

是的,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作为底层容器。

0