算法初步--集合求差（two pointers）

题目描述：

给定一个包含n个正整数的集合S1，再给定一个包含m个正整数的集合S2，求两个集合的差集，即S1−S2。

注：使用双指针法完成。

输入描述：

第一行两个正整数n、m（1≤n≤104,1≤m≤104），分别表示集合S1和S2的成员数；

第二行按升序给出S1的n个正整数成员（每个正整数的大小均不超过104，且互不相同）。

第三行按升序给出S2的m个正整数成员（每个正整数的大小均不超过104，且互不相同）。

输出描述：

按升序输出集合S1和S2的差集。结果之间用空格隔开，行末不允许有多余的空格。

样例：

输入：

5 3
1 2 5 6 8
2 4 6

输出：

1 5 8

解释

从集合S1中去掉集合S2中存在的整数2和6，还剩下{1,5,8}，即S1有但S2没有的，不包括S2有但S1没有的元素。

---

#include<stdio.h>
#define maxn 10001

int arr[maxn] = {};
int num = 0;

void getIntersection(int A[], int B[], int n, int m){
    int i = 0, j = 0;
    while(i < n && j < m){    // 相等，即A有B也有，不插入
        if(A[i] == B[j]){
            i++;
            j++;
        } 
        else if(A[i] > B[j]) j++;    // A没有B有，也不插入
        else arr[num++] = A[i++];    // A有B没有的，才插入
    }
    while(i < n) arr[num++] = A[i++];// 将A中可能没有遍历完的加入
}

int main(){
    int n, m;
    int arr1[maxn], arr2[maxn];
    scanf("%d%d", &n, &m);
    for(int i = 0; i < n; i++){
        scanf("%d", &arr1[i]);
    }
    for(int j = 0; j < m; j++){
        scanf("%d", &arr2[j]);
    }
    getIntersection(arr1, arr2, n, m);
    for(int i = 0; i < num; i++){
        printf("%d", arr[i]);
        if(i != num-1) printf(" ");
    }
    return 0;
}