C++ 学习之集合算法：set_intersection(交集)、set_union(并集)、set_difference(差集)

1、首先介绍一下相关概念

(1) 并集

对于两个给定集合A、B，由两个集合所有元素构成的集合，叫做A和B的并集。

记作：AUB  读作“A并B”

例：{ 3,5 }U{ 2,3,4,6 } = { 2,3,4,5,6 }

(2) 交集

对于两个给定集合A、B，由属于A又属于B的所有元素构成的集合，叫做A和B的交集。

记作： A∩B   读作“A交B”

例： A = { 1,2,3,4,5 }，B = { 3,4,5,6,8 }，A∩B = { 3,4,5 }

(3) 差集

假设有集合A和B，所有属于A且不属于B的元素的集合被称为A与B的差集。

示例：对于集合A = { a, b, c, d }和集合B = { b, c, w }，则A与B 的差集为{ a, d }

2、函数原型：

• set_intersection(iterator beg1, iterator end1, iterator beg2, iterator end2, iterator dest);

  // 求两个集合的交集

  // 注意:两个集合必须是有序序列

  // beg1 容器1开始迭代器
  // end1 容器1结束迭代器
  // beg2 容器2开始迭代器
  // end2 容器2结束迭代器
  // dest 目标容器开始迭代器

• set_union(iterator beg1, iterator end1, iterator beg2, iterator end2, iterator dest);

  // 求两个集合的并集

  // 注意:两个集合必须是有序序列

  // beg1 容器1开始迭代器
  // end1 容器1结束迭代器
  // beg2 容器2开始迭代器
  // end2 容器2结束迭代器
  // dest 目标容器开始迭代器

• set_difference(iterator beg1, iterator end1, iterator beg2, iterator end2, iterator dest);

  // 求两个集合的差集

  // 注意:两个集合必须是有序序列

  // beg1 容器1开始迭代器
  // end1 容器1结束迭代器
  // beg2 容器2开始迭代器
  // end2 容器2结束迭代器
  // dest 目标容器开始迭代器

3、示例代码

#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;

class print//输出数据的函数对象（仿函数）
{
public:
	void operator()(int v1)
	{
		cout << v1 << " ";
	}
};


/************************************* 求两个容器的交集：set_intersection ************************************************/

void test01()//求两个容器数据的交集，放到第三个容器中
{
	vector<int> v1;//两个集合必须是有序序列
	vector<int> v2;
	for (int i = 0; i < 10; i++)
	{
		v1.push_back(i);
		v2.push_back(i+2);
	}

	for_each(v1.begin(), v1.end(), print());//0 1 2 3 4 5 6 7 8 9
	cout << endl;
	for_each(v2.begin(), v2.end(), print());//2 3 4 5 6 7 8 9 10 11
	cout << endl;

	vector<int> target;
	//最特殊情况:一个容器完全包含另一个容器
	target.resize(min(v1.size(), v2.size()));//给目标容器分配空间，空间大小为两个源容器里边较小的一个


	//返回目标容器的最后一个元素的迭代器地址
	vector<int>::iterator itend = 
		set_intersection(v1.begin(), v1.end(), v2.begin(), v2.end(), target.begin());

	/*	
	//交集的大小一定是小于等于源容器里边较小的一个的，但是你分配的空间是等于较小容器的，resize分配的空间大于实际数据占用的空间，多余的会写0
	//所以这样写交集容器里边可能有几个无数据，被填充为0
	for_each(target.begin(), target.end(), print());//2 3 4 5 6 7 8 9 0 0 
	*/
	
	for_each(target.begin(), itend, print());// 2 3 4 5 6 7 8 9

}



/************************************* 求两个容器的并集：set_union ************************************************/

void test02()//求两个容器数据的并集，放到第三个容器中
{
	vector<int> v1;//两个集合必须是有序序列
	vector<int> v2;
	for (int i = 0; i < 10; i++)
	{
		v1.push_back(i);
		v2.push_back(i + 2);
	}

	for_each(v1.begin(), v1.end(), print());//0 1 2 3 4 5 6 7 8 9
	cout << endl;
	for_each(v2.begin(), v2.end(), print());//2 3 4 5 6 7 8 9 10 11
	cout << endl;

	vector<int> target;
	target.resize(v1.size() + v2.size());//给目标容器分配空间，空间大小为两个源容器里大小相加


	//返回目标容器的最后一个元素的迭代器地址
	vector<int>::iterator itend =
		set_union(v1.begin(), v1.end(), v2.begin(), v2.end(), target.begin());


	for_each(target.begin(), itend, print());//0 1 2 3 4 5 6 7 8 9 10 11 

}



/************************************* 求两个容器的差集：set_difference ************************************************/

void test03()//求两个容器数据的差集，放到第三个容器中
{
	vector<int> v1;//两个集合必须是有序序列
	vector<int> v2;
	for (int i = 0; i < 10; i++)
	{
		v1.push_back(i);
		v2.push_back(i + 3);
	}

	for_each(v1.begin(), v1.end(), print());//0 1 2 3 4 5 6 7 8 9
	cout << endl;
	for_each(v2.begin(), v2.end(), print());//3 4 5 6 7 8 9 10 11 12
	cout << endl;

	vector<int> target;
	//最特殊的情况，两个容器没有交集，那么差集就是较大的容器
	target.resize(max(v1.size(), v2.size()));//给目标容器分配空间，空间大小为两个源容器里较大的一个


	cout << "v1与v2的差集:" << endl;
	vector<int>::iterator itend =
		set_difference(v1.begin(), v1.end(), v2.begin(), v2.end(), target.begin());

	for_each(target.begin(), itend, print());//0 1 2 


	//target.clear();//清空之后，空间也释放了，需要重新分配空间
	//target.resize(max(v1.size(), v2.size()));
	//cout << endl;
	

	cout << "v2与v1的差集:" << endl;
	itend = set_difference(v2.begin(), v2.end(), v1.begin(), v1.end(), target.begin());

	for_each(target.begin(), itend, print());//10 11 12

}

int main()
{
	//test01();//set_intersection 取交集

	//test02();//set_union 取并集

	test03();//set_difference 取差集

	return 0;
}