c++容器 set 自定义排序
set c++ reference是c++中的一种容器,set可以用二叉树搜索树实现,set有两个特点:1. set中的元素不允许重复;2. set内部会维护一个严格的弱排序关系。上述两个特点实际上都依赖set的compare函数,compare函数判断两个元素相等就是相等,与元素本身没有直接的关系。默认定义了compare函数利用set内部默认的compare函数,可以将整数从小到大排序,将字符
·
set c++ reference是c++中的一种容器,set可以用二叉树搜索树实现,set有两个特点:
1. set中的元素不允许重复;
2. set内部会维护一个严格的弱排序关系。
上述两个特点实际上都依赖set的compare函数,compare函数判断两个元素相等就是相等,与元素本身没有直接的关系。
默认定义了compare函数
利用set内部默认的compare函数,可以将整数从小到大排序,将字符串按字母序进行排序。
#include <string>
#include <set>
#include <iostream>
using namespace std;
int main() {
int a[] = {20,10,30,40,50};
set<int> s1(a, a + 5);
for (auto it = s1.cbegin(); it != s1.cend(); it++)
{
cout << *it << " ";
}
cout << endl;
string b[] = {"apple", "banana", "pear", "orange", "strawberry"};
set<string> s2(b, b + 5);
for (auto it = s2.cbegin(); it != s2.cend(); it++)
{
cout << *it << " ";
}
cout << endl;
system("pause");
return 0;
}
输出:
10 20 30 40 50
apple banana orange pear strawberry
自定义排序函数
可以通过定义结构体(或类),并在其中重载()运算符,来自定义排序函数。然后,在定义set的时候,将结构体加入其中例如如下代码中的set<int, intComp>和set<string, strComp >。
#include <iostream>
#include <string>
#include <set>
using namespace std;
struct intComp {
bool operator() (const int& lhs, const int& rhs) const{
return lhs > rhs;
}
};
struct strComp
{
bool operator() (const string& str1, const string& str2) const {
return str1.length() < str2.length();
}
};
int main() {
int a[] = {10, 20, 30, 40, 50};
set<int, intComp> s1(a, a + 5);
for (auto it = s1.cbegin(); it != s1.cend(); it++)
{
cout << *it << " ";
}
cout << endl;
string b[] = {"apple", "banana", "pear", "orange", "strawberry"};
set<string, strComp > s2(b, b + 5);
for (auto it = s2.cbegin(); it != s2.cend(); it++)
{
cout << *it << " ";
}
cout << endl;
system("pause");
return 0;
}
输出:
50 40 30 20 10
pear apple banana strawberry
从上述程序段的输出可见,定义整型的比较函数后,输出数字按从大到小排序。定义字符串的比较函数为比较字符串的长度,输出字符串则按长度从小到大排序,其中并未出现orange了,因为banana的长度是6,而orange的长度也是6,set的构造函数认为orange与banana相同,而orange出现在后,因此不会有orange.
更多推荐
已为社区贡献2条内容
所有评论(0)