自定义STL容器比较函数可通过函数对象、函数指针或Lambda实现,用于控制排序行为。1. std::sort支持自定义比较规则,如降序或按结构体成员排序,推荐使用const引用避免拷贝;2. set/map通过模板参数传入比较器,可定义升序、降序或复杂逻辑(如Point坐标比较);3. priority_queue默认大根堆,需自定义比较器实现小根堆,如返回a>b创建最小堆。Lambda适用于简单场景,仿函数适合复杂或复用情况。关键在于比较函数返回true时表示第一个参数应排在第二个之前,逻辑需保持一致。

在c++中,自定义STL容器的比较函数通常用于控制排序行为或实现特定逻辑的元素顺序。常见场景包括 std::sort、std::set、std::map、std::priority_queue 等需要比较元素的容器或算法。可以通过函数对象(仿函数)、函数指针或Lambda表达式来实现。
1. 自定义 std::sort 的比较函数
对数组或vector等序列容器排序时,可通过传入比较函数改变默认升序规则。
例如,对整数降序排序:
#include <algorithm> #include <vector> #include <iostream> bool cmp(int a, int b) { return a > b; // 降序 } int main() { std::vector<int> vec = {3, 1, 4, 1, 5}; std::sort(vec.begin(), vec.end(), cmp); for (int x : vec) std::cout << x << " "; // 输出: 5 4 3 1 1 }
也可以使用Lambda:
立即学习“C++免费学习笔记(深入)”;
std::sort(vec.begin(), vec.end(), [](int a, int b) { return a > b; });
2. 自定义类类型的排序规则
若元素是自定义结构体,需明确如何比较。
例如按学生分数排序:
struct Student { std::string name; int score; }; std::vector<Student> students = {{"Alice", 85}, {"Bob", 90}, {"Charlie", 78}}; std::sort(students.begin(), students.end(), [](const Student& a, const Student& b) { return a.score > b.score; // 分数高者在前 });
注意:参数应使用const引用避免拷贝,提高效率。
3. 在 set 或 map 中使用自定义比较器
std::set 和 std::map 默认按键升序排列,若键为自定义类型或需不同顺序,需指定比较器作为模板参数。
例如,创建一个按降序排列的set:
struct greater_cmp { bool operator()(int a, int b) const { return a > b; } }; std::set<int, greater_cmp> s = {3, 1, 4, 1, 5}; // 遍历时输出: 5 4 3 1
对于结构体作为键的情况:
struct Point { int x, y; }; struct ComparePoint { bool operator()(const Point& a, const Point& b) const { if (a.x != b.x) return a.x < b.x; return a.y < b.y; } }; std::set<Point, ComparePoint> points;
4. 自定义 priority_queue 的比较方式
priority_queue 默认是大根堆(最大值优先),若要小根堆,需自定义比较器。
例如创建最小堆:
auto cmp = [](int a, int b) { return a > b; }; std::priority_queue<int, std::vector<int>, decltype(cmp)> pq(cmp); pq.push(3); pq.push(1); pq.push(4); // 顶部是1
或使用结构体:
struct MinHeap { bool operator()(int a, int b) { return a > b; // 小的优先级高 } }; std::priority_queue<int, std::vector<int>, MinHeap> pq;
基本上就这些。关键是理解比较函数返回true时表示第一个参数应排在第二个之前。在排序中返回a < b表示升序;在自定义容器中,逻辑一致即可。Lambda适合简单场景,结构体适合复杂或复用场景。不复杂但容易忽略细节。


