人见人爱A-B(set练习2)

原题链接

相似题型

Problem Description

参加过上个月月赛的同学一定还记得其中的一个最简单的题目，就是{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

思路：

• 题目要求输入两组数字，当输入第一组数字时，将其插入到 set 集合中。
• 当输入第二组数字时，使用 find() 函数查找集合中是否已经存在这个数，如果不存在，迭代器会指向集合末尾。如果第二组中的某个数字已经存在与集合中，则将其删除。
• 最后判断：如果集合中无元素，则输出NULL。否则定义一个迭代器，遍历并依次输出集合中的元素即可。

AC代码

#include <iostream>
#include <cstdio>
#include <set>

using namespace std;

int main()
{
    int n,m;
    while (cin >> n >> m)
    {
        //输入结束
        if (n == m && n == 0) break;
        
        set<int> s;
        
        int x;
        
        //输入第一组数字
        for (int i = 0;i < n;i++) 
        {
            cin >> x;
            s.insert(x);
        }
        
        //输入第二组数字并操作
        for (int i = 0;i < m;i++)
        {
            cin >> x;
            
            //find没有返回集合末尾,表示元素已存在
            if (s.find(x) != s.end()) s.erase(x);
        }
        
        //两组数字操作完成进行判断
        if (s.size() == 0) cout << "NULL" << endl;
        else 
        {
            //定义一个迭代器遍历set集合
            set<int>::iterator it;
            for (it = s.begin();it != s.end();it++) 
            {
                cout << *it << " ";
            }    
            cout << endl;
        }
    }
    return 0;
}