如何在C++ STL set中添加Lambda函数实现自定义排序规则

首先，你当前的代码存在两个核心问题：

1. 语法错误：不能直接将带捕获的Lambda作为set的模板参数。set的第二个模板参数要求是一个类型，而Lambda是匿名类型，且带捕获（[&]）的Lambda无法满足模板参数的类型要求。
2. 规则不符：你写的Lambda实现的是「int升序，int相同时字符串降序」，和你描述的需求不匹配（且int相同时int值本身相等，对应的规则没有实际意义，推测你实际想表达的是优先按字符串字典序降序排序，字符串相同时按int升序排序）。

下面给出几种正确实现自定义排序的方式：

方式一：使用不带捕获的Lambda + decltype #

不带捕获的Lambda可以被推导类型，我们先定义Lambda，再用decltype获取其类型作为set的模板参数，同时将Lambda传入set的构造函数：

#include <set>
#include <string>
#include <utility>
#include <iostream>

int main() {
    // 定义符合需求的Lambda比较器：优先字符串降序，字符串相同时int升序
    auto cmp = [](const std::pair<int, std::string>& a, const std::pair<int, std::string>& b) {
        if (a.second != b.second) {
            return a.second > b.second; // 字典序大的字符串排在前面
        } else {
            return a.first < b.first; // 字符串相同时，数值小的int排在前面
        }
    };

    // 用decltype获取Lambda类型，同时传入cmp作为构造参数
    std::set<std::pair<int, std::string>, decltype(cmp)> my_set(cmp);

    // 插入测试元素
    my_set.insert({1, "abc"});
    my_set.insert({2, "def"});
    my_set.insert({1, "def"});
    my_set.insert({3, "abc"});

    // 遍历输出验证顺序：(2,"def") → (1,"def") → (3,"abc") → (1,"abc")
    for (const auto& elem : my_set) {
        std::cout << elem.first << ", " << elem.second << std::endl;
    }

    return 0;
}

方式二：定义仿函数（函数对象类） #

这是兼容性更好的传统方式，适合需要复用比较规则的场景：

#include <set>
#include <string>
#include <utility>
#include <iostream>

// 自定义比较器类，重载()运算符实现比较逻辑
struct MyComparator {
    bool operator()(const std::pair<int, std::string>& a, const std::pair<int, std::string>& b) const {
        if (a.second != b.second) {
            return a.second > b.second;
        } else {
            return a.first < b.first;
        }
    }
};

int main() {
    // 直接使用自定义比较器类作为模板参数
    std::set<std::pair<int, std::string>, MyComparator> my_set;

    // 插入测试元素
    my_set.insert({1, "abc"});
    my_set.insert({2, "def"});
    my_set.insert({1, "def"});
    my_set.insert({3, "abc"});

    // 遍历输出验证
    for (const auto& elem : my_set) {
        std::cout << elem.first << ", " << elem.second << std::endl;
    }

    return 0;
}

方式三：使用std::function作为比较器类型 #

这种方式更灵活，适合需要动态切换比较规则的场景，但会带来轻微性能开销：

#include <set>
#include <string>
#include <utility>
#include <functional>
#include <iostream>

int main() {
    std::function<bool(const std::pair<int, std::string>&, const std::pair<int, std::string>&)> cmp =
        [](const auto& a, const auto& b) {
            if (a.second != b.second) {
                return a.second > b.second;
            } else {
                return a.first < b.first;
            }
        };

    std::set<std::pair<int, std::string>, decltype(cmp)> my_set(cmp);

    // 插入测试元素并验证
    my_set.insert({1, "abc"});
    my_set.insert({2, "def"});
    my_set.insert({1, "def"});
    my_set.insert({3, "abc"});

    for (const auto& elem : my_set) {
        std::cout << elem.first << ", " << elem.second << std::endl;
    }

    return 0;
}

关键注意点 #

自定义比较器必须满足严格弱序（Strict Weak Ordering），否则会导致set行为未定义：

• 不能同时存在cmp(a,b)和cmp(b,a)都为true的情况
• cmp(a,a)必须返回false

内容的提问来源于stack exchange，提问作者Codeforces Fan