算法初步--集合求差（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 2
2 2 2 6 8
2 6

输出：

2 2 8

解释

集合S1中有3个2，集合S2中有1个2，因此作差后还剩2个2；

集合S1中有1个6，集合S2中有1个6，因此作差后还剩0个6；

集合S1中有1个8，集合S2中有0个8，因此作差后还剩1个8；

因此差集为{2,2,8}

---

#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){
        if(A[i] == B[j]){        // 跟消消乐一样，相同的消去，都不插入
            i++;
            j++;
        } 
        else if(A[i] > B[j]) j++;    
        else arr[num++] = A[i++];    // A小，说明B中没有可以消去这个元素的，故插入
    }
    while(i < n) arr[num++] = A[i++];
}

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;
}