人见人爱A-B

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

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的意思是在A中存在的元素而在B中没有的,搞一搞就出来了
一开始我以为A为空和B为空要特殊说明 后来发现没有直接AC了,可能测试数据并没有把其设坑

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <memory.h>

using namespace std;

int main() {
    int arr3[101];
    int arr1[101],arr2[101],flag[101];
    int a,b;
    while (scanf("%d%d",&a,&b),a||b) {
//      if ((a==0)&&(b==0))
//      continue;

        for (int i=0; i<a; i++)
            scanf("%d",&arr1[i]); 

        for (int i = 0 ; i<b; i++)
            scanf("%d",&arr2[i]);

        memset(flag,0,sizeof(int)*101);

        for (int i = 0; i<a; i++)
            for (int j = 0; j<b; j++) {
                if (arr1[i]==arr2[j])
                    flag[i] = 1;
            }
        int t=0;
        for (int i = 0; i<a; i++) {
            if (!flag[i])
                arr3[t++]=arr1[i];
        }

        if (t==0) {
            printf("NULL\n");
            continue;
        }
        sort(arr3,arr3+t);
        for (int i = 0; i<t; i++)
            printf("%d ",arr3[i]);
        printf("\n");
    }
    return 0;
}