C/C++编程学习 - 第10周 ⑦ 人见人爱A-B

题目链接

题目描述

参加过上个月月赛的同学一定还记得其中的一个最简单的题目，就是{A}+{B}，那个题目求的是两个集合的并集，今天我们这个A-B求的是两个集合的差，就是做集合的减法运算。（当然，大家都知道集合的定义，就是同一个集合中不会有两个相同的元素，这里还是提醒大家一下）

呵呵，很简单吧？

Input
每组输入数据占1行,每行数据的开始是2个整数n(0<=n<=100)和m(0<=m<=100),分别表示集合A和集合B的元素个数，然后紧跟着n+m个元素，前面n个元素属于集合A，其余的属于集合B. 每个元素为不超出int范围的整数,元素之间有一个空格隔开.
如果n=0并且m=0表示输入的结束，不做处理。

Output
针对每组数据输出一行数据,表示A-B的结果,如果结果为空集合，则输出“NULL”,否则从小到大输出结果,为了简化问题，每个元素后面跟一个空格.

Sample Input

3 3 1 2 3 1 4 7
3 7 2 5 8 2 3 4 5 6 7 8 
0 0

Sample Output

2 3 
NULL

思路

求两个集合的差，我的思路是，先把输入进来的两个集合进行排序，然后对这两个集合的各个元素进行比较，如果某两个元素相等，那么就对被减的那个集合（假设为A集合）中的那个元素赋值为0，最后再对A集合排序，跳过值为0的数据，那么剩下的数据就是两个集合的差了。

C++代码：

#include<bits/stdc++.h>
using namespace std;
int A[200], B[200];
int main()
{
	int n, m;
	while(cin >> n >> m)
	{
		memset(A, 0, sizeof(A));
		memset(B, 0, sizeof(B));
		if(n == 0 && m == 0) break;
		for(int i = 0; i < n; i++)
			cin >> A[i];
		for(int i = 0; i < m; i++)
			cin >> B[i];
		sort(A, A + n);
		sort(B, B + m);
		for(int i = 0; i < n; i++)
			for(int j = 0; j < m; j++)
			{
				if(A[i] == B[j])
				{
					A[i] = 0;
					break;
				}
				if(A[i] < B[j]) break;
			}
		sort(A, A + n);
		int flag = 1;
		for(int i = 0; i < n; i++)
		{
			if(A[i] == 0) continue;
			flag = 0;
			cout << A[i] << " ";
		}
		if(flag) cout << "NULL";
		cout << endl;
	}
	return 0;
}