HDOJ 2034 人见人爱A-B

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

这个题意思就是数组A-数组B，输出余下的数。很简单，我用的是map关联器，输入a存入map，并设value=1，判断输入的b在a中是否存在，存在则修改value=2。最后输出value=1的所有key,若没有，那就输出null。

#include<iostream>
#include<map>

using namespace std;

int main()
{
    int n,m;
    while(cin>>n>>m&&n||m)
    {
        //初始化一个map:A
        map<int,int>A;
        int a,b,flag=0;
        //输入数组A
        for(int i=0;i<n;i++)
        {
            cin>>a;
            A[a]=1;
        }
        //输入数组B，并依次判断输入元素是否在A中
        for(int i=0;i<m;i++)
        {
            cin>>b;
            if(A.find(b)!=A.end())
            {
                A[b]=2;
                flag++;
            }
        }
        //当A为空输出NULL
        if(flag==n)
        {
            cout<<"NULL"<<endl;
        }
        else
        {
            //遍历容器，输出数组剩下的数值
           auto map_it=A.cbegin();
            while(map_it!=A.cend())
            {
                if(map_it->second==1)
                    cout<<map_it->first<<" ";
                ++map_it;
            }
            cout<<endl;
        }
    }
}