C++ 中的自定义排序函数，你学会了吗？（附详细代码）

本文链接：https://blog.youkuaiyun.com/2302_80970336/article/details/145521608

C++ 中的自定义排序函数，你学会了吗？

大家好！我是灰灰，今天咱们来聊聊 C++ 里 std::set 的自定义排序函数。都知道 std::set 是个很有用的关联容器，它会自动给元素排序。不过默认的排序方式有时候不太合咱心意，这时候就需要自定义排序函数出马啦！

一、函数对象（仿函数）来帮忙

啥是函数对象呢？简单来说，就是一个重载了 operator() 的类或者结构体。就好比你要开一家超市，默认的商品摆放是按价格从低到高（这就是 std::set 的默认排序），但你想按销量从高到低排，这就需要自定义排序啦。

#include <iostream>
#include <set>

// 自定义比较函数对象
struct CustomComparator {
    bool operator()(const int& a, const int& b) const {
        return a > b;  // 降序排序
    }
};

int main() {
    // 使用自定义比较函数对象创建 set
    std::set<int, CustomComparator> mySet = {1, 3, 2, 5, 4};

    // 遍历 set 并输出元素
    for (const auto& element : mySet) {
        std::cout << element << " ";
    }
    std::cout << std::endl;

    return 0;
}

在这段代码里，CustomComparator 就是那个帮你重新摆放商品的小助手。它重载的 operator() 就像是超市里重新整理货架的指令，告诉 std::set 要按降序来排列这些整数。咱在创建 std::set 的时候，把这个小助手CustomComparator作为第二个模板参数传进去，它就能按照咱们的要求来排序啦！

二、lambda 表达式闪亮登场

要是你觉得定义一个函数对象太麻烦，就想简单搞一搞，那 lambda 表达式绝对是你的心头好。就像你点外卖，不想自己做饭那么麻烦，直接下单就行。

#include <iostream>
#include <set>

int main() {
    // 定义 lambda 表达式作为比较函数
    auto customComp = [](const int& a, const int& b) {
        return a > b;  // 降序排序
    };

    // 使用 lambda 表达式创建 set
    std::set<int, decltype(customComp)> mySet(customComp);
    mySet.insert({1, 3, 2, 5, 4});

    // 遍历 set 并输出元素
    for (const auto& element : mySet) {
        std::cout << element << " ";
    }
    std::cout << std::endl;

    return 0;
}

这里的customComp就是那个简单的外卖订单，它用 lambda 表达式实现了降序排序的逻辑。咱在创建 std::set 的时候，用 decltype(customComp) 告诉它这个订单（比较函数）的类型，然后把订单传进去，就大功告成啦！

三、普通函数也能行

还有一种办法，就是用普通函数来作为比较函数。这就像是你有个老员工，一直按照你的老规矩来做事，现在你也可以让他来帮你重新整理超市货架。

#include <iostream>
#include <set>
#include <functional>

// 自定义比较函数
bool customCompare(const int& a, const int& b) {
    return a > b;  // 降序排序
}

int main() {
    // 使用普通函数创建 set
    std::set<int, std::function<bool(const int&, const int&)>> mySet(customCompare);
    mySet.insert({1, 3, 2, 5, 4});

    // 遍历 set 并输出元素
    for (const auto& element : mySet) {
        std::cout << element << " ";
    }
    std::cout << std::endl;

    return 0;
}