在std :: set中查找元素

时间:2015-08-12 18:51:10

标签: c++ c++11 stl stdset

鉴于我有一个std::set,我怎么知道该集合中的一个元素是否在另一个元素之前。例如,像这样 -

bool before(const std::set &s, const int first, const int second) const {
  auto it_first = s.find(first);
  auto it_second = s.find(second);
  return it_first <= it_second;
}

上面的代码不起作用,因为没有为双向迭代器定义<=,但是如何做这样的事情呢?

2 个答案:

答案 0 :(得分:7)

setoperator<(默认情况下)对其元素进行排序。可以通过key_compvalue_comp检索比较器本身。因此,如果两个元素都在集合中,则顺序由元素本身定义 - 您不需要迭代器:

return s.key_comp()(first, second);

如果集合中有一个或两个都没有,那么这取决于你在这些情况下想要做什么:

if (s.count(first)) {
    if (s.count(second)) {
        return s.key_comp()(first, second);
    }
    else {
        /* no second */
    }
}
else {
    /* no first, possibly second */
}

答案 1 :(得分:0)

如果您看起来比仅std::set更通用的解决方案,则此方法适用于具有forward迭代器的任何容器:

template <class T>
bool before( const T &cont, typename T::const_iterator first, typename T::const_iterator second )
{
    for( auto it = cont.begin(); true; ++it ) {
        if( it == first ) return true;
        if( it == second ) return false;
    }
 }

假设firstsecond是有效的迭代器。现在,您可以为std::setstd::map提供更优化的专业化:

return cont.key_comp()( *first, *second );

和具有随机访问迭代器的容器:

return first < second;