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

Question

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

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

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

Answer 1

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

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

Answer 2

Другой способ просто указать, существует ли элемент, - проверить 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
}

C ++ 20

В C ++ 20 set получает функцию contains, поэтому становится возможным следующее

if (myset.contains(x)) {
  // x is in the set
} else {
  // no x 
}

Answer 3

Просто для пояснения, причина, по которой в этих типах контейнеров нет таких элементов, как contains(), заключается в том, что это откроет вам возможность писать неэффективный код. Такой метод, вероятно, просто сделал бы внутреннее this->find(key) != this->end(), но подумайте, что вы делаете, когда ключ действительно присутствует; в большинстве случаев вы захотите получить элемент и что-то с ним сделать. Это означает, что вам нужно сделать секунду 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 ++ ...;)

Answer 4

Если вы собираетесь добавить функцию 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);
    }

}

Answer 5

Вы также можете проверить, находится ли элемент в наборе или нет при вставке элемента. Одноэлементная версия возвращает пару, для ее элемента pair :: first установлено значение итератора, указывающее либо на вновь вставленный элемент, либо на эквивалентный элемент, уже находящийся в наборе. Элемент pair :: second в паре устанавливается равным 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 будет указывать на позицию нового элемента в наборе.

Answer 6

В C ++ 20 мы наконец получим std::set::contains метод.

#include <iostream>
#include <string>
#include <set>

int main()
{
    std::set<std::string> example = {"Do", "not", "panic", "!!!"};

    if(example.contains("panic")) {
        std::cout << "Found\n";
    } else {
        std::cout << "Not found\n";
    }
}

Answer 7

Я использую

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

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

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

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

Answer 8

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

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

Answer 9

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

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

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

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.
 }

Answer 10

Мне удалось написать общую contains функцию для std::list и std::vector,

template<typename T>
bool contains( const list<T>& container, const T& elt )
{
  return find( container.begin(), container.end(), elt ) != container.end() ;
}

template<typename T>
bool contains( const vector<T>& container, const T& elt )
{
  return find( container.begin(), container.end(), elt ) != container.end() ;
}

// use:
if( contains( yourList, itemInList ) ) // then do something

Это немного очищает синтаксис.

Но я не смог использовать магический параметр шаблона шаблона , чтобы заставить эту работу работать с произвольными контейнерами stl.

// NOT WORKING:
template<template<class> class STLContainer, class T>
bool contains( STLContainer<T> container, T elt )
{
  return find( container.begin(), container.end(), elt ) != container.end() ;
}

Любые комментарии об улучшении последнего ответа были бы хорошими.