как найти пересечение двух std:: set в C++?
Я пытался найти пересечение между двумя std:: set в C++, но я продолжаю получать ошибку.
Я создал небольшой образец теста для этого
#include <iostream>
#include <vector>
#include <algorithm>
#include <set>
using namespace std;
int main() {
set<int> s1;
set<int> s2;
s1.insert(1);
s1.insert(2);
s1.insert(3);
s1.insert(4);
s2.insert(1);
s2.insert(6);
s2.insert(3);
s2.insert(0);
set_intersection(s1.begin(),s1.end(),s2.begin(),s2.end());
return 0;
}
последняя программа не генерирует никаких выходных данных, но я ожидаю, что у меня будет новый набор (назовем его s3
) со следующими значениями:
s3 = [ 1 , 3 ]
вместо этого я получаю ошибку:
test.cpp: In function ‘int main()’:
test.cpp:19: error: no matching function for call to ‘set_intersection(std::_Rb_tree_const_iterator<int>, std::_Rb_tree_const_iterator<int>, std::_Rb_tree_const_iterator<int>, std::_Rb_tree_const_iterator<int>)’
что я понимаю из этой ошибки, так это то, что в set_intersection
что принимает Rb_tree_const_iterator<int>
в качестве параметра.
кроме того, я полагаю std::set.begin()
метод возвращает объект такого типа,
есть ли лучший способ найти пересечение двух std::set
в C++? Предпочтительно встроенная функция?
Спасибо большое!
5 ответов:
вы не предоставили выходной итератор для set_intersection
template <class InputIterator1, class InputIterator2, class OutputIterator> OutputIterator set_intersection ( InputIterator1 first1, InputIterator1 last1, InputIterator2 first2, InputIterator2 last2, OutputIterator result );
исправить это, сделав что-то вроде
...; set<int> intersect; set_intersection(s1.begin(),s1.end(),s2.begin(),s2.end(), std::inserter(intersect,intersect.begin()));
вам понадобится
std::insert
итератор, так как набор на данный момент пуст. Мы не можем использовать back_ или front_inserter, так как set не поддерживает эти операции.
взгляните на образец в ссылке: http://en.cppreference.com/w/cpp/algorithm/set_intersection
вам нужен другой контейнер для хранения данных перекрестке, предположим, ниже код:
std::vector<int> common_data; set_intersection(s1.begin(),s1.end(),s2.begin(),s2.end(), std::back_inserter(common_data));
посмотреть std:: set_intersection. Вы должны добавить выходной итератор, где вы будете хранить результат:
#include <iterator> std::vector<int> s3; set_intersection(s1.begin(),s1.end(),s2.begin(),s2.end(), std::back_inserter(s3));
посмотреть Ideone для полного списка.
просто комментарий здесь. Я думаю, что пришло время добавить union, intersect operation в интерфейс set. Давайте предложим это в будущих стандартах. Я использую std в течение длительного времени, каждый раз, когда я использовал операцию набора, я хотел, чтобы std был лучше. Для некоторых сложных операций набора, таких как intersect, вы можете просто (проще?) измените следующий код:
template <class InputIterator1, class InputIterator2, class OutputIterator> OutputIterator set_intersection (InputIterator1 first1, InputIterator1 last1, InputIterator2 first2, InputIterator2 last2, OutputIterator result) { while (first1!=last1 && first2!=last2) { if (*first1<*first2) ++first1; else if (*first2<*first1) ++first2; else { *result = *first1; ++result; ++first1; ++first2; } } return result; }
скопировать из http://www.cplusplus.com/reference/algorithm/set_intersection/
например, если ваш вывод является набором, вы можете вывести.вставка(*first1). Кроме того, ваша функция не может быть шаблонизирована.Если код может быть короче, чем с помощью функции std set_intersection, то продолжайте его.
Если вы хотите сделать объединение двух наборов, вы можете просто установить.вставка(setB.begin (), setB.end ()); Это намного проще, чем метод set_union. Однако, это не будет работать с вектор.
первый (хорошо проголосованный) комментарий принято отвечать жалуется на отсутствие оператора для существующих операций набора std.
С одной стороны, я понимаю, что отсутствие таких операторов в стандартной библиотеке. С другой стороны, при желании их легко добавить (для личной радости). Я перегружен
operator *()
для пересечения множествoperator +()
для объединения множеств.образец
test-set-ops.cc
:#include <algorithm> #include <iterator> #include <set> template <class T, class CMP = std::less<T>, class ALLOC = std::allocator<T> > std::set<T, CMP, ALLOC> operator * ( const std::set<T, CMP, ALLOC> &s1, const std::set<T, CMP, ALLOC> &s2) { std::set<T, CMP, ALLOC> s; std::set_intersection(s1.begin(), s1.end(), s2.begin(), s2.end(), std::inserter(s, s.begin())); return s; } template <class T, class CMP = std::less<T>, class ALLOC = std::allocator<T> > std::set<T, CMP, ALLOC> operator + ( const std::set<T, CMP, ALLOC> &s1, const std::set<T, CMP, ALLOC> &s2) { std::set<T, CMP, ALLOC> s; std::set_union(s1.begin(), s1.end(), s2.begin(), s2.end(), std::inserter(s, s.begin())); return s; } // sample code to check them out: #include <iostream> using namespace std; template <class T> ostream& operator << (ostream &out, const set<T> &values) { const char *sep = " "; for (const T &value : values) { out << sep << value; sep = ", "; } return out; } int main() { set<int> s1 { 1, 2, 3, 4 }; cout << "s1: {" << s1 << " }" << endl; set<int> s2 { 0, 1, 3, 6 }; cout << "s2: {" << s2 << " }" << endl; cout << "I: {" << s1 * s2 << " }" << endl; cout << "U: {" << s1 + s2 << " }" << endl; return 0; }
собран и протестирован:
$ g++ -std=c++11 -o test-set-ops test-set-ops.cc $ ./test-set-ops s1: { 1, 2, 3, 4 } s2: { 0, 1, 3, 6 } I: { 1, 3 } U: { 0, 1, 2, 3, 4, 6 } $
что мне не нравится, так это копия возвращаемых значений в операторах. Может быть, это может быть решена с помощью перемещения назначения, но это все еще за пределами моих навыков.из-за моих ограниченных знаний об этой "новой причудливой" семантике перемещения я был обеспокоен возвращением оператора, которое может вызвать копии возвращенных наборов. Олаф Жи Dietsche указал, что эти опасения излишни, как
std::set
уже оснащен конструктором перемещения / назначением.хотя я верил ему, я думал, как это проверить (что-то вроде "самовнушение"). На самом деле, это довольно легко. Поскольку шаблоны должны быть предоставлены в исходном коде, вы можете просто пройти через отладчик. Таким образом, я поставил точку останова прямо на
return s;
наoperator *()
и продолжили с одним шагом, который привел меня сразу вstd::set::set(_myt&& _Right)
: et voilà-the move конструктор. Спасибо, Олаф, за (мое) просвещение.для полноты картины я также реализовал соответствующие операторы присваивания
operator *=()
для "разрушительного" пересечения множествoperator +=()
для "разрушительного" объединения множеств.пример
test-set-assign-ops.cc
:#include <iterator> #include <set> template <class T, class CMP = std::less<T>, class ALLOC = std::allocator<T> > std::set<T, CMP, ALLOC>& operator *= ( std::set<T, CMP, ALLOC> &s1, const std::set<T, CMP, ALLOC> &s2) { auto iter1 = s1.begin(); for (auto iter2 = s2.begin(); iter1 != s1.end() && iter2 != s2.end();) { if (*iter1 < *iter2) iter1 = s1.erase(iter1); else { if (!(*iter2 < *iter1)) ++iter1; ++iter2; } } while (iter1 != s1.end()) iter1 = s1.erase(iter1); return s1; } template <class T, class CMP = std::less<T>, class ALLOC = std::allocator<T> > std::set<T, CMP, ALLOC>& operator += ( std::set<T, CMP, ALLOC> &s1, const std::set<T, CMP, ALLOC> &s2) { s1.insert(s2.begin(), s2.end()); return s1; } // sample code to check them out: #include <iostream> using namespace std; template <class T> ostream& operator << (ostream &out, const set<T> &values) { const char *sep = " "; for (const T &value : values) { out << sep << value; sep = ", "; } return out; } int main() { set<int> s1 { 1, 2, 3, 4 }; cout << "s1: {" << s1 << " }" << endl; set<int> s2 { 0, 1, 3, 6 }; cout << "s2: {" << s2 << " }" << endl; set<int> s1I = s1; s1I *= s2; cout << "s1I: {" << s1I << " }" << endl; set<int> s2I = s2; s2I *= s1; cout << "s2I: {" << s2I << " }" << endl; set<int> s1U = s1; s1U += s2; cout << "s1U: {" << s1U << " }" << endl; set<int> s2U = s2; s2U += s1; cout << "s2U: {" << s2U << " }" << endl; return 0; }
собран и протестирован:
$ g++ -std=c++11 -o test-set-assign-ops test-set-assign-ops.cc $ ./test-set-assign-ops s1: { 1, 2, 3, 4 } s2: { 0, 1, 3, 6 } s1I: { 1, 3 } s2I: { 1, 3 } s1U: { 0, 1, 2, 3, 4, 6 } s2U: { 0, 1, 2, 3, 4, 6 } $