STL 应用之set
在以前学习stl的时候,曾经学到过,如果要将自定义的类型放入到set中的话,就需要重载“<”符号,原因是set是一个有序的集合,集合会按照“<”比较的大小,默认按照从小到大的顺序排列。假设我现在设计如下类型:
class mytype
{
public:
int a, b, c;
}
这是,为了让mytype类型可以顺利的放进set中,我必须重载“<”,这时问题来了,要如何重载呢?这个类型有三个数据成员,我能不能要求按照a的大小排列,如果a相等的话就随便按照b或者c的大小排列呢?如果近实现按照a的大小排列的话,重载函数如下:
bool operator<(const mytype& mytype) const
{
return a<mytype.a;
}
看起来简单明了,但是事实真的是这样吗?如果我声明一个set,并在其中放入mytype(1,2,3)、mytype(1,2,4)我能成功吗?实验了一下,结果如下:
测试时用的代码是这样的:
#include<iostream>
#include<set>
using namespace std;
class mytype
{
public:
int a, b, c;
mytype(int a, int b, int c):a(a), b(b), c(c){}
bool operator<(const mytype& mytype) const
{
return a<mytype.a;
}
};
int main()
{
set<mytype> se;
mytype type1(1,2,3);
mytype type2(1,2,4);
se.insert(type1);
se.insert(type2);
cout<<"the set size:"<<se.size()<<endl;
cout<<"elements in the set as follows:"<<endl;
for(set<mytype>::iterator it = se.begin(); it != se.end(); it++)
{
cout<<"("<<it->a<<", "<<it->b<<", "<<it->c<<") ";
}
cout<<endl;
return 0;
}
结果很明显,当我已经把mytype(1,2,3)放到set中之后,就不能把mytype(1,2,4)放进去了。但是为什么呢?这两个对象看起来确实不一样啊!stl在比较是否相同的时候不是要比较每一个数据成员的吗?从上述的例子中,看到的确实不是这样的,stl不会自动的做任何比较,它仅对你说明过的动作干一些指定的活儿。在重载“<”的时候,当只指定众多数据成员中的一部分的时候,如果这部分都相同的话,set就认为是同一个元素了。就如上述所示一样,重载的时候仅作了a的比较,而没有说明如果a相同的话,是否对剩下的元素进行比较。这样一来,set认为mytype(1,2,3)和mytype(1,2,4)是一样的。要让set正确的进行大小的比较,针对自定义类型,就必须说明所有的数据成员的比较情况。如上述的例子的“<”重载,应该这样写:
bool operator<(const mytype& mytype) const
{
return a<mytype.a?true:(b<mytype.b?true:c<mytype.c);
}
这样一来,就完全说明了每一个数据成员的比较关系,set就可以正常工作了。还是mytype(1,2,3)、mytype(1,2,4)两个元素,这回运行的结果如下:
运行代码为:
#include<iostream>
#include<set>
using namespace std;
class mytype
{
public:
int a, b, c;
mytype(int a, int b, int c):a(a), b(b), c(c){}
bool operator<(const mytype& mytype) const
{
return a<mytype.a?true:(b<mytype.b?true:c<mytype.c);
}
};
int main()
{
set<mytype> se;
mytype type1(1,2,3);
mytype type2(1,2,4);
se.insert(type1);
se.insert(type2);
cout<<"the set size:"<<se.size()<<endl;
cout<<"elements in the set as follows:"<<endl;
for(set<mytype>::iterator it = se.begin(); it != se.end(); it++)
{
cout<<"("<<it->a<<", "<<it->b<<", "<<it->c<<") ";
}
cout<<endl;
return 0;
}
摘自 martin's tips
上一篇: 快速计算32位数中1的位数
下一篇: LeetCode最常见的面试笔试题总结