c++

c++ 中的 set 怎自定义比较

小樊
81
2024-11-29 18:36:55
栏目: 编程语言

在 C++ 中,要自定义 set 的比较方式,您需要提供一个函数对象(也称为仿函数),该对象定义了元素之间的比较逻辑。这个函数对象应该接受两个参数并返回一个布尔值,表示这两个参数之间的大小关系。

以下是一个自定义比较函数的例子,用于创建一个按照字符串长度排序的 set

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

// 自定义比较函数对象
struct ByLength {
    bool operator()(const std::string& lhs, const std::string& rhs) const {
        return lhs.length() < rhs.length();
    }
};

int main() {
    // 使用自定义比较函数对象创建 set
    std::set<std::string, ByLength> my_set;

    // 向 set 中添加元素
    my_set.insert("apple");
    my_set.insert("banana");
    my_set.insert("cherry");
    my_set.insert("date");

    // 输出 set 中的元素
    for (const auto& elem : my_set) {
        std::cout << elem << std::endl;
    }

    return 0;
}

在这个例子中,我们定义了一个名为 ByLength 的结构体,其中包含一个名为 operator() 的成员函数。这个成员函数接受两个 std::string 类型的参数 lhsrhs,并根据它们的长度返回一个布尔值。然后,我们使用这个自定义比较函数对象 ByLength 来创建一个 std::set,并向其中添加一些字符串元素。最后,我们遍历并输出 set 中的所有元素。

0
看了该问题的人还看了