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

题目描述：

给定一个包含n个正整数的集合S1，再给定一个包含m个正整数的集合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 2 4 5 6 7 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]){        // 相等插入一个就行
            arr[num++] = A[i++];
            j++;
        } 
        else if(A[i] > B[j]){    // 不相等先插入小的
            arr[num++] = B[j++];
        }
        else{
            arr[num++] = A[i++];
        }
    }
    // 可能有没有遍历完的元素，需要如下处理
    while(i < n) arr[num++] = A[i++];
    while(j < m) arr[num++] = B[j++];
}

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