C++ STL - 벡터 끼리 연산 (합집합, 차집합, 교집합, 합병)

합집합 : merge() / set_union()

두 벡터를 붙힌다.  두 함수의 차이점이 있다. merge()는 중복 허용, set_union()은 중복 제거

 

Iterator iter 변수에 함수를 사용하고 결과의 끝을 가리키는 값을 반환받는다.

그 iter에서 벡터의 시작점을 빼서 값이 몇 개인지 알아내서 그 값으로 resize() 함수를 사용해서 벡터 크기 조절 후 순회

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

using namespace std;

int main(){

  // 반드시 정렬된 후 사용 가능
  vector<int> a = {1, 2, 3, 4, 5};
  vector<int> b = {5, 6, 7, 8, 9};
  vector<int> result1(10);
  vector<int> result2(10);
  vector<int>::iterator iter;
  
  // sort(a.begin(), a.end());
  // sort(b.begin(), b.end());
  
  // 머지 - [1, 2, 3, 4, 5, 5, 6, 7, 8, 9]
  merge(a, a+5, b, b+5, result1.begin());
  
  // 합집합 (A ∪ B) - [1, 2, 3, 4, 5, 6, 7, 8, 9]
  iter = set_union(a, a+5, b, b+5, result2.begin());
  result2.resize(iter - result2.begin());

  return 0;
}

 

차집합, 교집합, 대칭 차집합 : set_difference(), set_intersection(), set_symmetric_difference()

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

using namespace std;

int main(){

  // 반드시 정렬된 후 사용 가능
  vector<int> a = {1, 2, 3, 4, 5};
  vector<int> b = {5, 6, 7, 8, 9};
  vector<int> result1(10);
  vector<int> result2(10);
  vector<int> result3(10);
  vector<int> result4(10);
  vector<int>::iterator iter;
  
  // sort(a.begin(), a.end());
  // sort(b.begin(), b.end());
  
  // 차집합 (A - B) - [1, 2, 3, 4]
  iter = set_difference(a, a+5, b, b+5, result1.begin());
  result1.resize(iter - result1.begin());
  
  // 차집합 (B - A) - [6, 7, 8, 9]
  iter = set_difference(b, b+5, a, a+5, result2.begin());
  result2.resize(iter - result2.begin());
  
  // 교집합 (A ∩ B) - [5]
  iter = set_intersection(a, a+5, b, b+5, result3.begin());
  result3.resize(iter - result3.begin());
  
  // 대칭 차집합 (A △ B) - [1, 2, 3, 4, 6, 7, 8, 9] (교집합을 제외하는 느낌)
  iter = set_symmetric_difference(a, a+5, b, b+5, result4.begin());
  result4.resize(iter - result4.begin());
  
  return 0;
}