蓝桥杯 算法训练 集合运算

问题描述

给出两个整数集合A、B，求出他们的交集、并集以及B在A中的余集。
输入格式
　　第一行为一个整数n，表示集合A中的元素个数。
　　第二行有n个互不相同的用空格隔开的整数，表示集合A中的元素。
　　第三行为一个整数m，表示集合B中的元素个数。
　　第四行有m个互不相同的用空格隔开的整数，表示集合B中的元素。
　　集合中的所有元素均为int范围内的整数，n、m<=1000。
输出格式
　　第一行按从小到大的顺序输出A、B交集中的所有元素。
　　第二行按从小到大的顺序输出A、B并集中的所有元素。
　　第三行按从小到大的顺序输出B在A中的余集中的所有元素。
样例输入
5
1 2 3 4 5
5
2 4 6 8 10
样例输出
2 4
1 2 3 4 5 6 8 10
1 3 5
样例输入
4
1 2 3 4
3
5 6 7
样例输出
1 2 3 4 5 6 7
1 2 3 4

解题思路：

方法一：利用STL中set有序性与查找函数实现

#include<iostream>  
#include<algorithm>
#include<set>
#include<vector> 
using namespace std;
int A[1000] = {0};
int B[1000] = {0};
set<int> a;
set<int> b;
vector<int> inter;
set<int> unio;
vector<int> resi;

int main(){
	int m, n;
	cin >> m;
	for(int i = 0; i < m; i++){
		cin >> A[i];
		a.insert(A[i]);
	} 
	cin >> n;
	for(int i = 0; i < n; i++){
		cin >> B[i];
		b.insert(B[i]);
	} 
	//交集 
	set<int>::iterator it;
	for(it = a.begin(); it!=a.end(); it++){
		if(b.find(*it) != b.end()) inter.push_back(*it);
	} 
	//并集 
	for(it = a.begin(); it!=a.end(); it++){
		unio.insert(*it);
	}
	for(it = b.begin(); it!=b.end(); it++){
		unio.insert(*it);
	}
	//余集 
	for(it = a.begin(); it!=a.end(); it++){
		if(b.find(*it) == b.end()) resi.push_back(*it);
	} 
	
	//打印 
 	for(int i = 0; i < inter.size(); i++){
 		cout << inter[i] << " ";
	}
	cout << endl;
	for(it = unio.begin(); it != unio.end(); it++){
		cout << *it << " ";
	} 
	cout << endl;
	for(int i = 0; i < resi.size(); i++){
 		cout << resi[i] << " ";
	}
	cout << endl;
	return 0;
}

方法二：利用自带函数求解

#include<cstdio> 
#include<cstdlib> 
#include<cstring>
#include<iostream>  
#include<algorithm>
#include<set>
#include<vector> 
using namespace std;
vector<int> a;
vector<int> b;
vector<int> c;

void Print(vector<int> &c){
	for(int i=0;i<c.size();i++){
		cout<<c[i]<<" ";
	}
	c.clear();
	cout<<endl;
}

int main(){
	int m, n, temp;
	cin >> m;
	for(int i = 0; i < m; i++){
		cin >> temp;
		a.push_back(temp);
	}
	sort(a.begin(),a.end());
	cin >> n;
	for(int i = 0; i < n; i++){
		cin >> temp;
		b.push_back(temp);
	}
	sort(b.begin(),b.end());
	set_intersection(a.begin(),a.end(),b.begin(),b.end(),back_inserter(c));
	Print(c);
	set_union(a.begin(),a.end(),b.begin(),b.end(),back_inserter(c));
	Print(c); 
	set_difference(a.begin(),a.end(),b.begin(),b.end(),back_inserter(c));
	Print(c); 
}