C++中如何使用std::priority_queue自定义优先级

来源:建站教程作者:阿亮头衔:草根站长
导读:本期聚焦于小伙伴创作的《C++中如何使用std::priority_queue自定义优先级》,敬请观看详情,探索知识的价值。以下视频、文章将为您系统阐述其核心内容与价值。如果您觉得《C++中如何使用std::priority_queue自定义优先级》有用,将其分享出去将是对创作者最好的鼓励。

std::priority_queue是C++标准库中的优先队列容器,默认情况下它会按照元素从大到小的顺序出队,也就是大顶堆的实现。但实际开发中我们往往需要自定义优先级规则,比如实现小顶堆、按照自定义结构体的某个成员排序等,下面汇总几种常用的自定义优先级方法。

C++中如何使用std::priority_queue自定义优先级

方法一:重载结构体的比较运算符

如果优先队列存储的是自定义结构体,我们可以直接在结构体内部重载<运算符,来定义优先级规则。默认情况下优先队列使用<来判断优先级,所以重载后的规则会直接生效。

下面的示例定义了一个任务结构体,我们希望按照任务的优先级数值从小到大出队,也就是小顶堆的效果:

#include <iostream>
#include <queue>
#include <vector>

// 自定义任务结构体
struct Task {
    int id;
    int priority; // 优先级数值,越小优先级越高

    // 重载<运算符,定义优先级规则
    bool operator<(const Task& other) const {
        // 注意:priority_queue默认是大顶堆,所以这里要反过来写
        // 当当前任务的优先级大于其他任务时,当前任务优先级更低
        return priority > other.priority;
    }
};

int main() {
    std::priority_queue<Task> pq;
    pq.push({1, 3});
    pq.push({2, 1});
    pq.push({3, 2});

    while (!pq.empty()) {
        Task t = pq.top();
        pq.pop();
        std::cout << "任务ID: " << t.id << ", 优先级: " << t.priority << std::endl;
    }
    return 0;
}

运行上述代码,输出结果会按照优先级数值1、2、3的顺序出队,符合我们的预期。

方法二:使用仿函数作为比较器

如果不想修改结构体本身,或者需要多种不同的排序规则,可以自定义仿函数作为优先队列的比较器。仿函数本质是重载了()运算符的结构体或类。

下面的示例同样实现小顶堆的效果,但是不修改Task结构体:

#include <iostream>
#include <queue>
#include <vector>

struct Task {
    int id;
    int priority;
};

// 自定义仿函数,作为比较器
struct CompareTask {
    bool operator()(const Task& a, const Task& b) const {
        // 返回true表示a的优先级比b低
        return a.priority > b.priority;
    }
};

int main() {
    // 声明优先队列时传入比较器类型
    std::priority_queue<Task, std::vector<Task>, CompareTask> pq;
    pq.push({1, 3});
    pq.push({2, 1});
    pq.push({3, 2});

    while (!pq.empty()) {
        Task t = pq.top();
        pq.pop();
        std::cout << "任务ID: " << t.id << ", 优先级: " << t.priority << std::endl;
    }
    return 0;
}

方法三:使用lambda表达式作为比较器

C++11之后支持lambda表达式,我们可以用lambda来定义比较规则,这种方式更加灵活,适合临时定义简单的排序逻辑。

需要注意的是,lambda表达式作为比较器时,需要先在外部定义lambda变量,再传入优先队列的模板参数中:

#include <iostream>
#include <queue>
#include <vector>
#include <functional>

struct Task {
    int id;
    int priority;
};

int main() {
    // 定义lambda比较函数
    auto cmp = [](const Task& a, const Task& b) {
        return a.priority > b.priority;
    };
    // 声明优先队列,使用decltype获取lambda的类型
    std::priority_queue<Task, std::vector<Task>, decltype(cmp)> pq(cmp);
    pq.push({1, 3});
    pq.push({2, 1});
    pq.push({3, 2});

    while (!pq.empty()) {
        Task t = pq.top();
        pq.pop();
        std::cout << "任务ID: " << t.id << ", 优先级: " << t.priority << std::endl;
    }
    return 0;
}

方法四:内置类型的小顶堆实现

如果优先队列存储的是int、double等内置类型,想要实现小顶堆,不需要自定义结构体,可以直接使用标准库的std::greater比较器:

#include <iostream>
#include <queue>
#include <vector>
#include <functional>

int main() {
    // 小顶堆,使用std::greater作为比较器
    std::priority_queue<int, std::vector<int>, std::greater<int>> pq;
    pq.push(3);
    pq.push(1);
    pq.push(2);

    while (!pq.empty()) {
        std::cout << pq.top() << " ";
        pq.pop();
    }
    // 输出:1 2 3
    return 0;
}

不同方法的适用场景

我们可以根据需求选择合适的方法:

  • 如果自定义结构体只有一种固定的排序规则,优先选择重载比较运算符的方式,代码更简洁。
  • 如果需要多种不同的排序规则,或者不想修改结构体本身,选择仿函数的方式,可复用性更强。
  • 如果是临时使用的简单排序逻辑,选择lambda表达式的方式,代码更紧凑。
  • 如果是内置类型的小顶堆需求,直接使用std::greater即可,不需要额外定义比较规则。

注意事项

在使用自定义优先级时,需要注意优先队列的比较规则是:比较函数返回true时,第一个参数的优先级低于第二个参数。很多开发者会搞反这个逻辑,导致排序结果不符合预期。另外,自定义比较器时需要保证比较规则是严格弱序的,否则可能会出现未定义行为。

std::priority_queue自定义优先级比较函数仿函数lambda表达式修改时间:2026-07-13 21:12:26

免责声明:​ 已尽一切努力确保本网站所含信息的准确性。网站内容多为原创整理与精心编撰,观点力求客观中立。本站旨在免费分享,内容仅供个人学习、研究或参考使用。若引用了第三方作品,版权归原作者所有。如内容涉及您的权益,请联系我们处理。
内容垂直聚焦
专注技术核心技术栏目,确保每篇文章深度聚焦于实用技能。从代码技巧到架构设计,为用户提供无干扰的纯技术知识沉淀,精准满足专业提升需求。
知识结构清晰
覆盖从开发到部署的全链路。AI、前端、编程、数据库、服务器、建站、系统层层递进,构建清晰学习路径,帮助用户系统化掌握开发与运维所需的核心技术。
深度技术解析
拒绝泛泛而谈,深入技术细节与实践难点。无论是数据库优化还是服务器配置,均结合真实场景与代码示例进行剖析,致力于提供可直接应用于工作的解决方案。
专业领域覆盖
精准对应开发生命周期。从前端界面到后端编程,从数据库操作到服务器运维,形成完整闭环,一站式满足全栈工程师和运维人员的技术需求。
即学即用高效
内容强调实操性,步骤清晰、代码完整。用户可根据教程直接复现和应用于自身项目,显著缩短从学习到实践的距离,快速解决开发中的具体问题。
持续更新保障
专注既定技术方向进行长期、稳定的内容输出。确保各栏目技术文章持续更新迭代,紧跟主流技术发展趋势,为用户提供经久不衰的学习价值。