Как проверить, что элемент находится в std::set?
Как вы проверяете, что элемент находится в наборе?
Есть ли более простой эквивалент следующего кода:
myset.find(x) != myset.end()
12 ответов
Типичный способ проверить наличие во многих контейнерах 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(), что неэффективно. Лучше использовать поиск напрямую, чтобы вы могли кэшировать свой результат, например так:
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++...;)
В 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";
}
}
Если вы собираетесь добавить 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));
}
Редактировать:
Как указано в комментариях, я непреднамеренно использовал функцию, новую для 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);
}
}
Начиная с C++20 существует просто (и наконец!) Bool
std::contains(const K&) https://en.cppreference.com/w/cpp/container/set/contains
Вы также можете проверить, находится ли элемент в наборе или нет при вставке элемента. Версия с одним элементом возвращает пару, в которой для ее элемента 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 будет указывать на позицию нового элемента в наборе.
Напишите свой собственный:
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()) ....;
Моя версия только экономит мое время при написании кода. Я предпочитаю это так для конкурентного кодирования.
Я смог написать генерал 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() ;
}
Любые комментарии об улучшении последнего ответа были бы хорошими.
Это на милю.
bool once(uintptr_t val) {
return visited.emplace(val).second;
}
Как может быть иначе?
https://godbolt.org/z/9zP77jqMc
func5(unsigned long):
sub rsp, 24
mov QWORD PTR [rsp+8], rdi
lea rsi, [rsp+8]
mov edi, OFFSET FLAT:visited2
call std::pair<std::_Rb_tree_iterator<unsigned long>, bool> std::_Rb_tree<unsigned long, unsigned long, std::_Identity<unsigned long>, std::less<unsigned long>, std::allocator<unsigned long> >::_M_emplace_unique<unsigned long&>(unsigned long&)
add rsp, 24
mov eax, edx
ret
// общий синтаксис
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.
}