东华OJ 基础题55 人见人爱A-B

问题描述 :

A和B是两个集合，A-B求的是两个集合的差，就是做集合的减法运算。（当然，大家都知道集合的定义，就是同一个集合中不会有两个相同的元素，这里还是提醒大家一下）呵呵，很简单吧？

输入说明 :

输入数据包含T个测试实例。

首先输入数字T，然后输入T组测试数据，每组输入数据占1行,每行数据的开始是2个整数n(0<=n<=100)和m(0<=m<=100),分别表示集合A和集合B的元素个数，然后紧跟着n+m个元素，前面n个元素属于集合A，其余的属于集合B. 每个元素为不超出int范围的整数,元素之间由一个空格隔开.

输出说明 :

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

输入范例 :

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

输出范例 :

2 3 
NULL

完整C++代码

代码看上去好像有点复杂，其实就是分了几种情况讨论

1. 遍历完M数组，N数组还有元素的情况——继续遍历N数组输出
2. 创建临时变量count记录两个数组中相同的元素，如果和此时结束时的i相同，则认为N数组已经遍历完全且在M数组中都存在

#include <bits/stdc++.h>

using namespace std;

int main()
{
	int T;
	cin >> T;
	while (T--) {
		int n, m;
		cin >> n >> m;
		int arrN[n], arrM[m];
		for (int i = 0; i < n; i++) {
			cin >> arrN[i];
		}
		for (int i = 0; i < m; i++) {
			cin >> arrM[i];
		}
		sort(arrN, arrN + n);
		sort(arrM, arrM + m);
		int i = 0, j = 0, count = 0;
		for (; i < n && j < m;) {
			if (arrN[i] < arrM[j]) {
				cout << arrN[i] << " ";
				i++;
			} else if (arrN[i] > arrM[j]) {
				j++;
			} else {
				i++;j++;count++;
			}
		}
		if (j == m && i < n) {
			while (i < n) {
				cout << arrN[i++] << " ";
			}
		} else if (count == i) {
			cout << "NULL" << endl;
		} else {
			cout << endl;
		}
		
	}
	return 0;
}

😋欢迎大伙私信或者评论区交流讨论😋