как найти пересечение двух 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 в интерфейс 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, продолжайте.

Если вы хотите сделать объединение двух наборов, вы можете просто setA.вставка(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 }

$