修改stl::set相关源码，提供有序属性值的查找接口

普通的stl::set，查找时只能传入key_type。 不能使用属性值查找。

例如:

/* an employee record holds its ID, name and age */
class employee
{
public:
  int         id;
  std::string name;
  int         age;
public:
  employee():id(0){}
  employee(int id_,std::string name_,int age_):id(id_),name(name_),age(age_){}
  employee(const employee& e2)
  {
      *this = e2;
  }

  friend std::ostream& operator<<(std::ostream& os,const employee& e)
  {
    os<<e.id<<" "<<e.name<<" "<<e.age<<std::endl;
    return os;
  }
  bool operator<(const employee& e2) const
  {
      return this->id < e2.id;
  }

  employee& operator=(const employee& e2)
  {
      id = e2.id;
      name = e2.name;
      age = e2.age;
      return *this;
  }

};

stl::set<employee> set1;

set1.insert(employee(2,"Aristotle",23));
set1.insert(employee(3,"Albert",20));
set1.insert(employee(4,"John",57));

set1::iterator it;
it = set1.find(employee(3,"Albert",20));   //find参数必须是employee对象，不能只提供id进行查找

 

可以自定义一个带模板的find函数。stl是开源的，可以自己拿出相关代码进行改动，不使用编译库自带的。

 

在stl_set.h中原来的find()函数相同的位置 添加带模板的find函数。

//针对属性值的查找接口，需要自定义比较函数。只能在set的当前索引下有序的属性查找，即_T必须是set中有序的属性。stl的find接口只能接受key_type对象作为参数
      template<class _T, class _T_Compare>
      const_iterator
      find(const _T& __x, _T_Compare com)
      {   return _M_t.find(__x, com);  }

 

在stl_rdtree.h中find()函数相同的位置 添加带模板的find函数，来提供上层set需要调用的接口。

    //提供针对属性值的lower_bound接口，使用template
    template<class _T, class _T_Compare>
    iterator
        lower_bound(const _T& __k, _T_Compare com)
        {  return _M_lower_bound(_M_begin(), _M_end(), __k, com); }

    //提供针对属性值的查找接口，使用template
    template<class _T, class _T_Compare>
    const_iterator
        find(const _T& __k, _T_Compare com) 
        {
            const_iterator __j = _M_lower_bound(_M_begin(), _M_end(), __k, com);
            return (__j == end()
                    || com(__k, 
                        _S_key(__j._M_node))) ? end() : __j;
        }

 

这样就可以使用

     set1::iterator it = set1.find(3);  //查找id为3的对象