温馨提示×

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

c++
小亿
105
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