求两集合的交集

题目

有两个集合采用递增有序的整数顺序表A，B存储，设计一个在时间上尽可能高效的算法，求两个集合的交集C，C仍然用顺序表存储，并给出算法的时间和空间复杂度。

算法描述

1. 初始化指针：为 𝐴A 和 𝐵B 设置两个指针 𝑖i 和 𝑗j，初始都指向第一个元素。
2. 遍历两个列表：
   • 比较 𝐴[𝑖]A[i] 和 𝐵[𝑗]B[j]：
     • 如果 𝐴[𝑖]=𝐵[𝑗]A[i]=B[j]，这意味着找到了一个交集中的元素，将其加入到 𝐶C 中，然后 𝑖i 和 𝑗j 都递增。
     • 如果 𝐴[𝑖]<𝐵[𝑗]A[i]<B[j]，递增 𝑖i，因为 𝐴[𝑖]A[i] 不可能与 𝐵B 中的任何更大的元素相匹配。
     • 如果 𝐴[𝑖]>𝐵[𝑗]A[i]>B[j]，递增 𝑗j，因为 𝐵[𝑗]B[j] 不可能与 𝐴A 中的任何更大的元素相匹配。
3. 继续：重复以上步骤直到 𝐴A 或 𝐵B 的任一数组遍历完毕。

代码

#include <iostream>

void intersection(int A[], int nA, int B[], int nB, int C[], int& nC) {
    int i = 0, j = 0;
    nC = 0;  // 初始化交集的大小为0

    while (i < nA && j < nB) {
        if (A[i] == B[j]) {
            C[nC++] = A[i];  // 存储交集元素
            i++;
            j++;
        } else if (A[i] < B[j]) {
            i++;
        } else {
            j++;
        }
    }
}

int main() {
    int A[] = {1, 3, 5, 7, 9};
    int nA = sizeof(A) / sizeof(A[0]);
    int B[] = {3, 4, 5, 6, 7};
    int nB = sizeof(B) / sizeof(B[0]);
    int C[100];  // 交集数组，假设足够大
    int nC;

    intersection(A, nA, B, nB, C, nC);

    std::cout << "Intersection of A and B:" << std::endl;
    for (int i = 0; i < nC; ++i) {
        std::cout << C[i] << " ";
    }
    std::cout << std::endl;

    return 0;
}

• 时间复杂度：𝑂(𝑛+𝑚)O(n+m)，其中 𝑛n 是列表 𝐴A 的长度，𝑚m 是列表 𝐵B 的长度。每个列表只被遍历一次，操作的次数是两个列表的长度之和。
• 空间复杂度：𝑂(𝑚𝑖𝑛(𝑛,𝑚))O(min(n,m))，这是因为在最坏的情况下，交集的大小可能等于较小数组的大小。