蓝桥杯 ALGO-10 集合运算

  算法训练 集合运算  

时间限制：1.0s   内存限制：512.0MB

      

问题描述

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

分析：用map在输入时存储分类结果，按要求输出结果。

代码：

#include<iostream>
#include<algorithm>
#include<map>
using namespace std;
int main() {
	int n, m;
	map<int, int> mp;
	cin >> n;
	int temp;
	for (int i = 0; i < n; i++) {
		cin >> temp;
		mp[temp] = 1;
	}
	cin >> m;
	for (int i = 0; i < m; i++) {
		cin >> temp;
		if (mp[temp] == 1) {
			mp[temp] = 2;
		} else {
			mp[temp] = 3;
		}
	}
	int flag = 0;
	for (map<int, int>::iterator it = mp.begin(); it != mp.end(); it++) {
		if (it->second == 2) {
			if (flag == 0) {
				cout << it->first;
				flag = 1;
			} else {
				cout << ' ' << it->first;
			}
		}
	}
	if (flag == 1) {
		cout << endl;
	}
	flag = 0;
	for (map<int, int>::iterator it = mp.begin(); it != mp.end(); it++) {
		if (it->second != 0) {
			if (flag == 0) {
				cout << it->first;
				flag = 1;
			} else {
				cout << ' ' << it->first;
			}
		}
	}
	if (flag == 1) {
		cout << endl;
	}
	flag = 0;
	for (map<int, int>::iterator it = mp.begin(); it != mp.end(); it++) {
		if (it->second == 1) {
			if (flag == 0) {
				cout << it->first;
				flag = 1;
			} else {
				cout << ' ' << it->first;
			}
		}
	}
	return 0;
}