自定义的类型放入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)我能成功吗?实验了一下,结果如下:

image

测试时用的代码是这样的:

#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)两个元素,这回运行的结果如下:

image

运行代码为:

#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;
}

 

原文转自 http://www.cnblogs.com/malloc/archive/2012/03/13/2394719.html

posted @ 2016-04-23 18:19  ~小小鸟~  阅读(2486)  评论(1编辑  收藏  举报