两个数组的交集：双指针与集合的实现方法

在编程中，求两个数组的交集是一个常见的操作。对于这个问题，开发者通常会采用不同的方法来实现。以下将详细介绍两种常用的实现方法：一种是利用双指针进行逐个比较，另一种是结合集合数据结构来存储结果。每种方法都有其独特的优缺点。

双指针法：高效的O(n log n)时间复杂度

双指针法是解决这个问题的经典方法之一。这种方法的核心思想是对两个数组进行排序，然后使用两个指针分别从两个数组的开头开始移动。当两个指针指向的元素相等时，说明这个元素就是交集的一部分，可以加入结果集合中，同时移动两个指针。若左边的元素小于右边的，则左指针前进；反之，则右指针前进。当其中一个指针达到数组末尾时，循环结束。

这种方法的时间复杂度主要由排序操作决定，为O(n log n)，因为两个数组都需要进行一次排序操作。空间复杂度方面，除了存储结果外，其他额外空间复杂度为O(1)。这种方法在实际应用中表现优异，尤其是在处理较大的数组时，效率较高。

代码实现：

public int[] intersect(int[] nums1, int[] nums2) {    Arrays.sort(nums1);    Arrays.sort(nums2);    HashSet
   
     set = new HashSet<>();    int i = 0, j = 0;    while (i < nums1.length && j < nums2.length) {        if (nums1[i] == nums2[j]) {            set.add(nums1[i]);            i++;            j++;        } else if (nums1[i] < nums2[j]) {            i++;        } else {            j++;        }    }    return set.stream().mapToInt(Integer::intValue).toArray();}

集合法：更直观的实现方式

另一种实现方法是使用集合来存储交集元素。这种方法的思路是将两个数组中的元素都存储到一个集合中，然后通过遍历其中一个数组，检查每个元素是否也存在在另一个数组中。如果存在，则将这个元素加入结果集合中。这种方法的时间复杂度为O(n + m)，其中n和m分别是两个数组的长度。空间复杂度方面，除了存储结果外，其他额外空间复杂度为O(n)。

这种方法的优点是实现起来相对简单，读者可以轻松理解和修改。然而，集合的查询操作可能会带来一定的性能上的开销，尤其是在处理非常大的数据集时。

代码实现：

public int[] intersection(int[] nums1, int[] nums2) {    ArrayList
   
     list = new ArrayList<>();    for (int num : nums1) {        list.add(num);    }    ArrayList
    
      result = new ArrayList<>();    for (int num : nums2) {        if (list.contains(num) && !result.contains(num)) {            result.add(num);        }    }    return result.stream().mapToInt(Integer::intValue).toArray();}

双指针法改进版：直接存储结果

在前面的双指针法实现中，我们使用了集合来存储交集元素。虽然这种方法效率较高，但有一些情况下可能会因为集合的额外查询开销而影响性能。为了优化这种情况，可以直接使用一个列表来存储结果。这种方法可以减少一些不必要的查询操作，同时保持相同的时间复杂度。

代码实现：

public int[] intersect_2(int[] nums1, int[] nums2) {    Arrays.sort(nums1);    Arrays.sort(nums2);    ArrayList
   
     list = new ArrayList<>();    int i = 0, j = 0;    while (i < nums1.length && j < nums2.length) {        if (nums1[i] == nums2[j]) {            list.add(nums1[i]);            i++;            j++;        } else if (nums1[i] < nums2[j]) {            i++;        } else {            j++;        }    }    return list.stream().mapToInt(Integer::intValue).toArray();}

优化方法：复用双指针实现

为了进一步优化，我们可以将两个数组排序后，直接使用一个指针来遍历其中一个数组，同时在另一个数组中查找对应的元素。这种方法的时间复杂度仍然是O(n log n)，但实现起来稍微复杂一些。

代码实现：

public int[] intersection2(int[] nums1, int[] nums2) {    Set
   
     set = new HashSet<>();    Arrays.sort(nums1);    Arrays.sort(nums2);    int i = 0, j = 0;    while (i < nums1.length && j < nums2.length) {        if (nums1[i] == nums2[j]) {            set.add(nums1[i]);            i++;            j++;        } else if (nums1[i] < nums2[j]) {            i++;        } else {            j++;        }    }    int[] res = new int[set.size()];    int index = 0;    for (int num : set) {        res[index++] = num;    }    return res;}

总结

以上是两个数组交集问题的几种常见实现方法。每种方法都有其独特的优点和适用场景。选择哪种方法取决于具体的应用需求。如果需要在时间复杂度上做进一步的优化，可以选择双指针法；如果需要实现起来简单，可以选择集合法或者直接存储结果的方法。无论选择哪种方法，理解其工作原理都是非常重要的，这将有助于你在面对更复杂的问题时做出更明智的选择。

转载地址：http://ccltz.baihongyu.com/

你可能感兴趣的文章