Как узнать, присутствует ли элемент в std :: vector?

616

Все, что я хочу сделать, это проверить, существует ли элемент в векторе или нет, чтобы я мог разобраться с каждым случаем.

if ( item_present )
   do_this();
else
   do_that();
Джоан Венге
источник
2
поиск в векторе очень медленный, так как вы должны смотреть на каждый элемент вектора, поэтому рассмотрите возможность использования карты, если вы выполняете много поисков
naumcho
7
@naumcho: если вектор отсортирован, всегда есть бинарный поиск, как показано ниже. Это делает его таким же быстрым, как карта, и если вы храните только значения (а не карты ключ / значение), тогда он будет использовать намного меньше памяти.
Адам Хос
4
Карты, конечно, не лучший выбор, но использование набора может быть полезным. Если вам нужно время поиска O (1), лучше использовать hash_set.
Филипп
Превосходный ответ на дублирующий вопрос: stackoverflow.com/a/3451045/472647
CodeMouse92
1
Если вы собираетесь искать разные числа несколько раз, хеш-таблица будет более эффективной.
NL628

Ответы:

916

Вы можете использовать std::findот <algorithm>:

#include <vector>
vector<int> vec; 
//can have other data types instead of int but must same datatype as item 
std::find(vec.begin(), vec.end(), item) != vec.end()

Это возвращает bool ( trueесли присутствует, в falseпротивном случае). С вашим примером:

#include <algorithm>
#include <vector>

if ( std::find(vec.begin(), vec.end(), item) != vec.end() )
   do_this();
else
   do_that();
MSN
источник
216
Я не понимаю, как count () может быть быстрее, чем find (), поскольку find () останавливается, как только найден один элемент, а count () всегда должен сканировать всю последовательность.
Эрик Маленфант
114
Не забывайте об этом, #include <algorithm>иначе вы можете получить очень странные ошибки, такие как «не могу найти подходящую функцию в пространстве имен std»
rustyx
80
Разве это никого не беспокоило, что, несмотря на то, что STL является «объектно-ориентированным», .find()он все еще не является функцией-членом std::vector, как и следовало ожидать? Интересно, это как-то является следствием шаблонов?
Бобобобо
71
@bobobobo: ООП не имеет ничего общего с членами или не членами. И существует широко распространенное мнение о том, что если что-то не обязательно должно быть членом или когда оно не дает никаких преимуществ при реализации в качестве члена, то оно не должно быть членом; std::vector<>::find()не даст никакого преимущества, и при этом он не нужен, поэтому нет, он не должен быть членом. См. Также en.wikipedia.org/wiki/Coupling_%28computer_programming%29
Себастьян Мах
36
@phresnel Я бы сказал, что «когда он не дает никаких преимуществ при реализации в качестве члена», в этом случае ложь. Преимущество заключается в упрощенном и понятном интерфейсе. Например: mvec.find(key) != mvec.cend()предпочтительнее std::find(mvec.cbegin(), mvec.cend(), key) != mvec.cend().
Суалог
113

Как уже говорили другие, используйте STL findили find_ifфункции. Но если вы ищете в очень больших векторов , и это влияет на производительность, вы можете сортировать вектор , а затем использовать binary_search, lower_boundили upper_boundалгоритмы.

Брайан Нил
источник
3
Хороший ответ! Найти всегда о (п). lower_bound равен o (log (n)), если используется с итераторами с произвольным доступом.
Стивен Эдмондс
30
Сортировка - это O (nlogn), поэтому она стоит только если вы выполняете больше, чем O (logn) поиск.
Лиори
7
@liori Правда, это зависит от ваших моделей использования. Если вам нужно отсортировать его только один раз, а затем многократно выполнять поиск, это может спасти вас.
Брайан Нил
1
@ Брайан Нил, сортировка большого вектора стоит того, чтобы на нем было много элементов поиска. Сортировка будет O (nlogn), а O (n) будет лучше, если нужно найти элемент только один раз :)
Swapnil B.
47

Используйте find из заголовка алгоритма stl. Я проиллюстрировал его использование с типом int. Вы можете использовать любой понравившийся вам тип, если вы можете сравнить на равенство (перегрузка ==, если вам нужно для вашего пользовательского класса).

#include <algorithm>
#include <vector>

using namespace std;
int main()
{   
    typedef vector<int> IntContainer;
    typedef IntContainer::iterator IntIterator;

    IntContainer vw;

    //...

    // find 5
    IntIterator i = find(vw.begin(), vw.end(), 5);

    if (i != vw.end()) {
        // found it
    } else {
        // doesn't exist
    }

    return 0;
}
м-диез
источник
2
В зависимости от потребностей OP, find_if () также может быть подходящим. Это позволяет искать, используя произвольный предикат вместо равенства.
Эрик Маленфант
К сожалению, ваш комментарий слишком поздно. Ответ, который я дал, также упоминает find_if.
Фрэнк
39

Если ваш вектор не упорядочен, используйте подход, предложенный MSN:

if(std::find(vector.begin(), vector.end(), item)!=vector.end()){
      // Found the item
}

Если ваш вектор упорядочен, используйте метод binary_search, предложенный Брайаном Нилом:

if(binary_search(vector.begin(), vector.end(), item)){
     // Found the item
}

Бинарный поиск дает O (log n) производительность в худшем случае, что намного эффективнее, чем первый подход. Чтобы использовать бинарный поиск, вы можете использовать qsort, чтобы сначала отсортировать вектор, чтобы гарантировать его упорядоченность.

spiralmoon
источник
3
Ты имеешь в виду std::sort? qsortочень неэффективно для векторов .... см .: stackoverflow.com/questions/12308243/…
Джейсон Р. Мик
1
Бинарный поиск будет работать лучше для больших контейнеров, но для небольших контейнеров простой линейный поиск, вероятно, будет таким же быстрым или быстрым.
BillT
21

Я использую что-то вроде этого ...

#include <algorithm>


template <typename T> 
const bool Contains( std::vector<T>& Vec, const T& Element ) 
{
    if (std::find(Vec.begin(), Vec.end(), Element) != Vec.end())
        return true;

    return false;
}

if (Contains(vector,item))
   blah
else
   blah

... так оно и есть на самом деле понятно и читаемо. (Очевидно, вы можете повторно использовать шаблон в нескольких местах).

Энди Кроувел
источник
и вы можете заставить его работать для списков или векторов, используя 2 typenames
Erik Aronesty
@ErikAronesty вы можете обойтись с 1 аргументом шаблона, если вы используете value_typeиз контейнера для типа элемента. Я добавил ответ как этот.
Мартин Бродхерст
13

В C ++ 11 вы можете использовать any_of. Например, если это vector<string> v;тогда:

if (any_of(v.begin(), v.end(), bind(equal_to<string>(), _1, item)))
   do_this();
else
   do_that();

В качестве альтернативы используйте лямбду:

if (any_of(v.begin(), v.end(), [&](const std::string& elem) { return elem == item; }))
   do_this();
else
   do_that();
Дэцин
источник
1
bind1stи bind2ndявляются устаревшими , так как C ++ 11 и полностью удал ли в C ++ 17. Используйте bindс placeholdersи / или лямбды вместо этого.
Андре
11

Вот функция, которая будет работать для любого контейнера:

template <class Container> 
const bool contains(const Container& container, const typename Container::value_type& element) 
{
    return std::find(container.begin(), container.end(), element) != container.end();
}

Обратите внимание, что вы можете обойтись без 1 параметра шаблона, потому что вы можете извлечь value_typeиз контейнера. Вам нужно, typenameпотому что Container::value_typeэто зависимое имя .

Мартин Бродхерст
источник
5
Обратите внимание, что это иногда слишком широко - например, это работает для std :: set, но дает ужасную производительность по сравнению с функцией-членом find (). Я нашел, что лучше всего добавить специализацию для контейнеров с более быстрым поиском (set / map, unordered_ *)
Andy Krouwel
10

Имейте в виду, что, если вы собираетесь выполнять много поисков, есть контейнеры STL, которые лучше подходят для этого. Я не знаю, какое у вас приложение, но стоит рассмотреть такие ассоциативные контейнеры, как std :: map.

std :: vector является контейнером выбора, если у вас нет причины для другого, и поиск по значению может быть такой причиной.

Дэвид Торнли
источник
Даже при поиске по значению вектор может быть хорошим выбором, если он отсортирован и вы используете binary_search, lower_bound или upper_bound. Если содержимое контейнера изменяется между поисками, вектор не очень хорош из-за необходимости повторной сортировки.
Ренце де Ваал
8

Используйте функцию поиска STL .

Имейте в виду, что есть также функция find_if , которую вы можете использовать, если ваш поиск более сложный, т.е. если вы не просто ищете элемент, но, например, хотите посмотреть, есть ли элемент, который удовлетворяет определенному условие, например, строка, которая начинается с "abc". ( find_ifдаст вам итератор, который указывает на первый такой элемент).

Фрэнк
источник
7

С надстройкой вы можете использовать any_of_equal:

#include <boost/algorithm/cxx11/any_of.hpp>

bool item_present = boost::algorithm::any_of_equal(vector, element);
Михаил
источник
5

Вы можете попробовать этот код:

#include <algorithm>
#include <vector>

// You can use class, struct or primitive data type for Item
struct Item {
    //Some fields
};
typedef std::vector<Item> ItemVector;
typedef ItemVector::iterator ItemIterator;
//...
ItemVector vtItem;
//... (init data for vtItem)
Item itemToFind;
//...

ItemIterator itemItr;
itemItr = std::find(vtItem.begin(), vtItem.end(), itemToFind);
if (itemItr != vtItem.end()) {
    // Item found
    // doThis()
}
else {
    // Item not found
    // doThat()
}
TrungTN
источник
3

Вы можете использовать findфункцию, найденную в stdпространстве имен, то есть std::find. Вы передаете std::findфункцию, beginи endитератор из вектора вы хотите найти, наряду с элементом вы ищете и сравнить полученный итератор к концу вектора , чтобы увидеть , если они совпадают или нет.

std::find(vector.begin(), vector.end(), item) != vector.end()

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

TankorSmash
источник
3

Вы можете использовать счет тоже. Он вернет количество элементов, присутствующих в векторе.

int t=count(vec.begin(),vec.end(),item);
Адитья
источник
11
findбыстрее, чем count, потому что он не продолжает считать после первого матча.
Камиль Гудесюн
2

Если вы хотите найти строку в векторе:

    struct isEqual
{
    isEqual(const std::string& s): m_s(s)
    {}

    bool operator()(OIDV* l)
    {
        return l->oid == m_s;
    }

    std::string m_s;
};
struct OIDV
{
    string oid;
//else
};
VecOidv::iterator itFind=find_if(vecOidv.begin(),vecOidv.end(),isEqual(szTmp));
Gank
источник
2

Еще один пример с использованием операторов C ++.

#include <vector>
#include <algorithm>
#include <stdexcept>

template<typename T>
inline static bool operator ==(const std::vector<T>& v, const T& elem)
{
  return (std::find(v.begin(), v.end(), elem) != v.end());
}

template<typename T>
inline static bool operator !=(const std::vector<T>& v, const T& elem)
{
  return (std::find(v.begin(), v.end(), elem) == v.end());
}

enum CODEC_ID {
  CODEC_ID_AAC,
  CODEC_ID_AC3,
  CODEC_ID_H262,
  CODEC_ID_H263,
  CODEC_ID_H264,
  CODEC_ID_H265,
  CODEC_ID_MAX
};

void main()
{
  CODEC_ID codec = CODEC_ID_H264;
  std::vector<CODEC_ID> codec_list;

  codec_list.reserve(CODEC_ID_MAX);
  codec_list.push_back(CODEC_ID_AAC);
  codec_list.push_back(CODEC_ID_AC3);
  codec_list.push_back(CODEC_ID_H262);
  codec_list.push_back(CODEC_ID_H263);
  codec_list.push_back(CODEC_ID_H264);
  codec_list.push_back(CODEC_ID_H265);

  if (codec_list != codec)
  {
    throw std::runtime_error("codec not found!");
  }

  if (codec_list == codec)
  {
    throw std::logic_error("codec has been found!");
  }
}
Valdemar_Rudolfovich
источник
4
Я бы не советовал злоупотреблять перегрузкой операторов таким способом.
Леон
2
Леон, я согласен с тобой, семантически это не правильно. Я использую его, чтобы сделать модульные тесты более четкими.
Вальдемар_Рудольфович
1
template <typename T> bool IsInVector(T what, std::vector<T> * vec)
{
    if(std::find(vec->begin(),vec->end(),what)!=vec->end())
        return true;
    return false;
}
user3157855
источник
1

(C ++ 17 и выше):

можно использовать std::searchтакже

Это также полезно для поиска последовательности элементов.

#include <algorithm>
#include <iostream>
#include <vector>

template <typename Container>
bool search_vector(const Container& vec, const Container& searchvec)
{
    return std::search(vec.begin(), vec.end(), searchvec.begin(), searchvec.end()) != vec.end();
}

int main()
{
     std::vector<int> v = {2,4,6,8};

     //THIS WORKS. SEARCHING ONLY ONE ELEMENT.
     std::vector<int> searchVector1 = {2};
     if(search_vector(v,searchVector1))
         std::cout<<"searchVector1 found"<<std::endl;
     else
         std::cout<<"searchVector1 not found"<<std::endl;

     //THIS WORKS, AS THE ELEMENTS ARE SEQUENTIAL.
     std::vector<int> searchVector2 = {6,8};
     if(search_vector(v,searchVector2))
         std::cout<<"searchVector2 found"<<std::endl;
     else
         std::cout<<"searchVector2 not found"<<std::endl;

     //THIS WILL NOT WORK, AS THE ELEMENTS ARE NOT SEQUENTIAL.
     std::vector<int> searchVector3 = {8,6};
     if(search_vector(v,searchVector3))
         std::cout<<"searchVector3 found"<<std::endl;
     else
         std::cout<<"searchVector3 not found"<<std::endl;
}

Также есть гибкость прохождения некоторых поисковых алгоритмов. Обратитесь сюда.

https://en.cppreference.com/w/cpp/algorithm/search

Паван Чандака
источник
1

В последнее время я лично использовал шаблоны для обработки нескольких типов контейнеров, а не только для работы с векторами. Я нашел аналогичный пример в Интернете (не помню, где), так что заслуга принадлежит тому, от кого я это украл. Этот конкретный шаблон, похоже, также обрабатывает необработанные массивы.

template <typename Container, typename T = typename std::decay<decltype(*std::begin(std::declval<Container>()))>::type>
bool contains(Container && c, T v)
{
    return std::find(std::begin(c), std::end(c), v) != std::end(c);
}
Паскаль Лаферриер
источник
-4

Используя Newton C ++, это проще, самодокументировано и быстрее, чем с std :: find, потому что возвращает bool напрямую.

bool exists_linear( INPUT_ITERATOR first, INPUT_ITERATOR last, const T& value )

bool exists_binary( INPUT_ITERATOR first, INPUT_ITERATOR last, const T& value )

Я думаю, что очевидно, что делают функции.

include <newton/algorithm/algorithm.hpp>

if ( newton::exists_linear(first, last, value) )
   do_this();
else
   do_that();
Мойзес Рохо
источник