在 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
中的所有元素。