蓝桥杯试题 算法训练 集合运算

问题链接

“蓝桥杯”练习系统 (lanqiao.cn)

问题描述

给出两个整数集合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

思路

直接调用C++函数库

代码

#include <stdio.h>
//#include <bits/stdc++.h>
#include <algorithm>
#include <set>
#include <iostream>
using namespace std;

int main()
{
	set<int> A;
	set<int> B;
	set<int> intersection; //交集 
	set<int> bunion; //并集 
	set<int> difference; //差集 
	set<int> chaji;
	set<int>::iterator i;
	int n, m;
	cin >> n;
	while (n --)
	{
		int i;
		cin >> i; 
		A.insert(i);	
	} 
	cin >> m;
	while (m --)
	{
		int i;
		cin >> i;
		B.insert(i);
	}
	
	set_intersection(A.begin(), A.end(), B.begin(), B.end(), inserter(intersection, intersection.begin()));
	set_union(A.begin(), A.end(), B.begin(), B.end(), inserter(bunion, bunion.begin()));
	set_difference(A.begin(), A.end(), B.begin(), B.end(), inserter(chaji, chaji.begin()));
	
	//交集 
	for ( i =intersection.begin(); i != intersection.end(); i ++)
	{
		cout << *i << " ";
	}
	cout << endl; 
	//并集
	for ( i =bunion.begin(); i != bunion.end(); i ++)
	{
		cout << *i << " ";
	}
	cout << endl;
	//B在A中的余集中的所有元素 
	for ( i =chaji.begin(); i != chaji.end(); i ++)
	{
		cout << *i << " ";
	}
	
	
	return 0;
}