7-4 集合减法

给定两个非空集合A和B，集合的元素为30000以内的正整数，编写程序求A-B。

输入格式:

输入为三行。第1行为两个整数n和m，分别为集合A和B包含的元素个数，1≤n, m ≤10000。第2行表示集合A，为n个空格间隔的正整数，每个正整数不超过30000。第3行表示集合B，为m个空格间隔的正整数，每个正整数不超过30000。

输出格式:

输出为一行整数，表示A-B，每个整数后一个空格，各元素按递增顺序输出。若A-B为空集，则输出0，0后无空格。

输入样例:

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

输出样例:

1 2 

代码：
 

#include<bits/stdc++.h>
using namespace std;
#define OK 1
#define ERROR 0
#define OVERFLOW -2
int a[30002];//空间换时间
int b[30002];
typedef int ElemType;
typedef int Status;
typedef struct
{
    ElemType *elem;
    int length;
}Sqlist;

Status Init_Sq(Sqlist &S,Sqlist &D)//
{
    int num1=S.length;
    int num2=D.length;
    S.elem=(ElemType*)malloc(sizeof(ElemType)*num1);
    D.elem=(ElemType*)malloc(sizeof(ElemType)*num2);
    for(int i=0;i<num1;i++)
    {
        cin>>S.elem[i];//输入进线性表
        a[S.elem[i]]=1;//进入数组
    }
    for(int i=0;i<num2;i++)
    {
        cin>>D.elem[i];
        b[D.elem[i]]=1;
    }
    return OK;
}
int main()
{
    Sqlist S,D;
    cin>>S.length>>D.length;
    if(Init_Sq(S,D))
    {
        int flag=0;
        for(int i=0;i<S.length;i++)
        {
            if(a[S.elem[i]]&&!b[S.elem[i]])//只要S中有D中没有就可以
            {
                cout<<S.elem[i]<<" ";
                flag=1;
            }

        }
        if(!flag)cout<<"0"<<endl;//空集直接输出0
    }
}