acm 2034

人见人爱A-B

Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 65536/32768 K (Java/Others) Total Submission(s): 43070    Accepted Submission(s): 12081

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

 

Author

lcy

 

#include <map>
#include <iostream>

using namespace std;


int main()
{

    map<int,bool> setMapA;
    int n,m,num;
    while(cin >> n >> m)
    {
        if(m == 0 && n == 0)
        {
            break;
        }
        setMapA.clear();
        for(int i = 0; i< n;i++)
        {
            cin >> num;
            setMapA.insert(pair<int,bool>(num,true));
        }

        for(int i = 0; i< m;i++)
        {
            cin >> num;
            setMapA[num] = false;
        }

        for(map<int,bool>::iterator cmp = setMapA.begin(); cmp != setMapA.end();)
        {
            if(!(cmp->second))
            {
                setMapA.erase(cmp++);
            }else
            {
                ++cmp;
            }
        }

        if(setMapA.size() == 0)
        {
            cout<<"NULL
";
        }else
        {
            int i = 0;
            for(map<int,bool>::iterator cmp = setMapA.begin(); cmp != setMapA.end();cmp++)
            {
                if(i != (int)setMapA.size() -1)
                {
                    cout << cmp->first << " ";
                }else
                {
                    cout << cmp->first << " 
";
                }
                i++;
            }

        }


    }
}

map中使用erase函数，需要注意setMapA.erase(cmp++);这样先将cmp所指节点信息删除了，再将cmp节点向前指一个；
如果setMapA.erase(++cmp);删除了下个节点的信息，后面链断了的 情况，从而造成bug