在 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
类型的参数 lhs
和 rhs
,并根据它们的长度返回一个布尔值。然后,我们使用这个自定义比较函数对象 ByLength
来创建一个 std::set
,并向其中添加一些字符串元素。最后,我们遍历并输出 set
中的所有元素。