Как проверить, что элемент находится в std:: set?

Как проверить, что элемент находится в наборе?

есть ли более простой эквивалент следующего кода:

myset.find(x) != myset.end()

9 ответов


типичный способ проверить наличие во многих контейнерах STL:

const bool is_in = container.find(element) != container.end();

другой способ просто сказать, существует ли элемент, - это проверить count()

if (myset.count(x)) {
   // x is in the set, count is 1
} else {
   // count zero, i.e. x not in the set
}

так что мне все равно придется найти итератор. Тогда, конечно, лучше просто сравнить его с end тоже.

set< X >::iterator it = myset.find(x);
if (it != myset.end()) {
   // do something with *it
}

просто чтобы уточнить, почему нет такого члена, как contains() в этих типах контейнеров, потому что это откроет вас для написания неэффективного кода. Такой метод, вероятно, просто сделает this->find(key) != this->end() внутренне, но подумайте, что вы делаете, когда ключ действительно присутствует; в большинстве случаев вы захотите получить элемент и сделать что-то с ним. Это означает, что вам придется сделать второй find(), что неэффективно. Лучше использовать find напрямую, чтобы вы могли кэшировать свой результат, например Итак:

Container::const_iterator it = myContainer.find(key);
if (it != myContainer.end())
{
  // Do something with it, no more lookup needed.
}
else
{
  // Key was not present.
}

конечно, если вы не заботитесь об эффективности, вы всегда можете свернуть свой собственный, но в этом случае вы, вероятно, не следует быть на C++... ;)


если вы собирались добавить contains функция, это может выглядеть так:

#include <algorithm>
#include <iterator>

template<class TInputIterator, class T> inline
bool contains(TInputIterator first, TInputIterator last, const T& value)
{
    return std::find(first, last, value) != last;
}

template<class TContainer, class T> inline
bool contains(const TContainer& container, const T& value)
{
    // This works with more containers but requires std::begin and std::end
    // from C++0x, which you can get either:
    //  1. By using a C++0x compiler or
    //  2. Including the utility functions below.
    return contains(std::begin(container), std::end(container), value);

    // This works pre-C++0x (and without the utility functions below, but doesn't
    // work for fixed-length arrays.
    //return contains(container.begin(), container.end(), value);
}

template<class T> inline
bool contains(const std::set<T>& container, const T& value)
{
    return container.find(value) != container.end();
}

работает с std::set, другие контейнеры STL, и даже массивы фиксированной длины:

void test()
{
    std::set<int> set;
    set.insert(1);
    set.insert(4);
    assert(!contains(set, 3));

    int set2[] = { 1, 2, 3 };
    assert(contains(set2, 3));
}

Edit:

как указано в комментариях, я непреднамеренно использовал функцию, новую для C++0x (std::begin и std::end). Вот почти тривиальная реализация от VS2010:

namespace std {

template<class _Container> inline
    typename _Container::iterator begin(_Container& _Cont)
    { // get beginning of sequence
    return (_Cont.begin());
    }

template<class _Container> inline
    typename _Container::const_iterator begin(const _Container& _Cont)
    { // get beginning of sequence
    return (_Cont.begin());
    }

template<class _Container> inline
    typename _Container::iterator end(_Container& _Cont)
    { // get end of sequence
    return (_Cont.end());
    }

template<class _Container> inline
    typename _Container::const_iterator end(const _Container& _Cont)
    { // get end of sequence
    return (_Cont.end());
    }

template<class _Ty,
    size_t _Size> inline
    _Ty *begin(_Ty (&_Array)[_Size])
    { // get beginning of array
    return (&_Array[0]);
    }

template<class _Ty,
    size_t _Size> inline
    _Ty *end(_Ty (&_Array)[_Size])
    { // get end of array
    return (&_Array[0] + _Size);
    }

}

вы также можете проверить, является ли элемент в наборе или нет при вставке элемента. Единственный элемент вариант возврата пары с парой членов::первый набор для итератор, указывающий на либо вставленный элемент или эквивалентные элементы уже в комплекте. Второй элемент пары::имеет значение true, если новый элемент был вставлен, или false, если эквивалентный элемент уже существовал.

например: предположим, что набор уже 20 как элемент.

 std::set<int> myset;
 std::set<int>::iterator it;
 std::pair<std::set<int>::iterator,bool> ret;

 ret=myset.insert(20);
 if(ret.second==false)
 {
     //do nothing

 }
 else
 {
    //do something
 }

 it=ret.first //points to element 20 already in set.

если элемент недавно вставлен, чем pair:: first будет указывать на положение нового элемента в наборе.


напишите свой собственный:

template<class T>
bool checkElementIsInSet(const T& elem, const std::set<T>& container)
{
  return container.find(elem) != container.end();
}

Я использую

if(!my_set.count(that_element)) //Element is present...
;

но это не так эффективно, как

if(my_set.find(that_element)!=my_set.end()) ....;

моя версия экономит мое время только при написании кода. Я предпочитаю такой способ конкурентного кодирования.


Я могу написать общие


//общий синтаксис

       set<int>::iterator ii = find(set1.begin(),set1.end(),"element to be searched");

/ * в приведенном ниже коде я пытаюсь найти элемент 4 in и int set, если он присутствует или нет*/

set<int>::iterator ii = find(set1.begin(),set1.end(),4);
 if(ii!=set1.end())
 {
    cout<<"element found";
    set1.erase(ii);// in case you want to erase that element from set.
 }