STL算法——常用集合算法（set_intersection、set_union、set_difference）

5.6 常用集合算法

算法简介：

• set_intersection //求两个容器的交集
• set_union //求两个容器的并集
• set_difference //求两个容器的差集

5.6.1 set_intersection

功能描述：

• 求两个容器的交集

函数原型：

• set_intersection(iterator beg1, iterator end1, iterator beg2, iterator end2, iterator dest);
  • 求两个集合的交集
  • 注意：两个集合必须是有序序列
  • beg1 容器1开始迭代器
  • end1 容器1结束迭代器
  • beg2 容器2开始迭代器
  • end2 容器2结束迭代器
  • dest 目标容器开始迭代器

5.6.2 set_union

功能描述：

• 求两个集合的并集

函数原型：

• set-union(iterator beg1, iterator end1, iterator beg2, iterator end2, iterator dest)
  • 求两个集合的并集
  • 注意：两个集合必须是有序序列
  • beg1 容器1开始迭代器
  • end1 容器1结束迭代器
  • beg2 容器2开始迭代器
  • end2 容器2结束迭代器
  • dest 目标容器开始迭代器

5.6.3 set_difference

功能描述：

• 求两个集合的差集

函数原型：

• set-fidderence(iterator beg1, iterator end1, iterator beg2, iterator end2, iterator dest)
  • 求两个集合的差集
  • 注意：两个集合必须是有序序列
  • beg1 容器1开始迭代器
  • end1 容器1结束迭代器
  • beg2 容器2开始迭代器
  • end2 容器2结束迭代器
  • dest 目标容器开始迭代器

5.6.4 示例

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

//打印输出
class myPrint {
public:
	void operator()(int val) {
		cout << val << " ";
	}
};

//求两个容器的交集

void test01() {
	vector<int> v1;
	v1.push_back(1);
	v1.push_back(2);
	v1.push_back(3);
	v1.push_back(4);

	vector<int> v2;
	v2.push_back(3);
	v2.push_back(4);
	v2.push_back(5);
	v2.push_back(6);

	cout << "v1: ";
	for_each(v1.begin(), v1.end(), myPrint());
	cout << endl;

	cout << "v2: ";
	for_each(v2.begin(), v2.end(), myPrint());
	cout << endl;

	//目标容器提前开辟空间
	vector<int> vTarget1;
	//最特殊的情况是是大容器包含小容器
	vTarget1.resize(min(v1.size(), v2.size()));

	//获取交集
	vector<int>::iterator itEnd1 =  set_intersection(v1.begin(), v1.end(), v2.begin(), v2.end(), vTarget1.begin());
	cout << "v1与v2的交集：";
	for_each(vTarget1.begin(), itEnd1, myPrint());
	cout << endl;

	//获取并集
	vector<int> vTarget2;
	//最特殊的情况是每个容器的值都不相同
	vTarget2.resize(v1.size() + v2.size());
	vector<int>::iterator itEnd2 = set_union(v1.begin(), v1.end(), v2.begin(), v2.end(), vTarget2.begin());
	cout << "v1与v2的并集：";
	for_each(vTarget2.begin(), itEnd2, myPrint());
	cout << endl;
	
	//获取差集
	vector<int> vTarget3;
	//最特殊的情况是每个容器的值都不相同
	vTarget3.resize(max(v1.size(), v2.size()));
	vector<int>::iterator itEnd3 = set_difference(v1.begin(), v1.end(), v2.begin(), v2.end(), vTarget3.begin());
	cout << "v1与v2的差集：";
	for_each(vTarget3.begin(), itEnd3, myPrint());
	cout << endl;

}

//主函数
int main() {
	test01();
	system("pause");
	return 0;
}