试题 算法训练 P0705

输入两个整数集合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

#include<iostream>
using namespace std;

int main() {
	int n, m;
	int max = 0;
	int num[10000] = {0};
	
	cin >> n;
	
	for (int i = 0; i < n; i ++) {
		int temp;
		
		cin >> temp;
		
		if (temp > max) {
			max = temp;
		}
		
		num[temp] += 1;
	}
	
	cin >> m;
	
	for (int i = 0; i < m; i ++) {
		int temp;
		
		cin >> temp;
		
		if (temp > max) {
			max = temp;
		}
		
		num[temp] += 2;
	}
	
	for (int i = 0; i <= max; i ++) {
		if (num[i] == 3) {
			cout << i << " ";
		}
	}
	
	cout << endl;
	
	for (int i = 0; i <= max; i ++) {
		if (num[i] != 0) {
			cout << i << " ";
		}
	}
	
	cout << endl;
	
	for (int i = 0; i <= max; i ++) {
		if (num[i] == 1) {
			cout << i << " ";
		}
	}
	
	return 0;
} 

总结：

如果是A，就对应那个数+1

如果是B，就+2

交集就直接求，值为3的（A+B）

并集求不为0的

B在A中的余集就是值为1（A）