您的位置:

Java取交集详解

一、概述

在编程中,取交集(Intersection)是一个经常使用的操作。在Java中,我们可以使用不同的数据结构和算法来实现取交集的功能。本文将详细介绍Java取交集的实现方式,包括数组、集合和Stream等不同的方案,并且给出完整的代码示例。

二、数组实现取交集

数组是Java中最基本的数据结构之一,也是最简单的实现取交集的方式。

实现方式:

public static int[] intersect(int[] nums1, int[] nums2) {
    Arrays.sort(nums1);
    Arrays.sort(nums2);
    int i = 0, j = 0, k = 0;
    while (i < nums1.length && j < nums2.length) {
        if (nums1[i] < nums2[j]) {
            i++;
        } else if (nums1[i] > nums2[j]) {
            j++;
        } else {
            nums1[k++] = nums1[i++];
            j++;
        }
    }
    return Arrays.copyOfRange(nums1, 0, k);
}

以上代码先将两个数组进行排序,然后使用双指针移动比较得出交集。

三、集合实现取交集

Java中的集合也可以很方便地实现取交集。

实现方式:

public static Set<Integer> intersect(Set<Integer> set1, Set<Integer> set2) {
    Set<Integer> result = new HashSet<>();
    for (Integer integer : set1) {
        if (set2.contains(integer)) {
            result.add(integer);
        }
    }
    return result;
}

以上代码使用了Java中的Set类,通过contains方法判断两个集合是否相交,并将结果存储在一个新的Set中。

四、Stream实现取交集

Java 8中的新特性Stream也可以实现取交集的功能。

实现方式:

public static Set<Integer> intersect(int[] nums1, int[] nums2) {
    Set<Integer> set1 = Arrays.stream(nums1).boxed().collect(Collectors.toSet());
    Set<Integer> set2 = Arrays.stream(nums2).boxed().collect(Collectors.toSet());
    return set1.stream().filter(set2::contains).collect(Collectors.toSet());
}

以上代码将数组转换为Set,然后通过Stream中的filter方法筛选出相交的元素,最后将结果存储在一个新的Set中。

五、总结

本文介绍了Java取交集的三种实现方式,分别是使用数组、集合和Stream。通过比较这三种方式,我们可以发现不同的数据结构和算法会影响程序的实现效率和易用性。在实际开发中,我们需要结合具体的场景来选择最合适的方式。