Java中计算两个数组的交集

文
章
目
录

• 1.使用HashSet求数组交集
• 2.使用流(Streams)求数组交集

学习如何在Java中使用HashSet类找到两个数组的交集。交集是指同时存在于两个数组中的共同项。

例如，在array1和array2中，交集将包含元素{4,5}。

Integer[] array1 = new Integer[]{1, 2, 3, 4, 5};
Integer[] array2 = new Integer[]{4, 5, 6, 7};
//交集 : {4, 5}

1.使用HashSet求数组交集

找到数组的交集 要获取两个数组的交集，请按照以下步骤操作：

• 将第一个数组推入一个HashSet中。
• 使用retainAll()方法保留只存在于第二个数组中的元素。

以下是一个Java程序，用于获取两个整数数组之间的交集并打印输出。

Integer[] array1 = new Integer[]{1, 2, 3, 4, 5};
Integer[] array2 = new Integer[]{4, 5, 6, 7};

HashSet<Integer> set = new HashSet<>();
set.addAll(Arrays.asList(array1));
set.retainAll(Arrays.asList(array2));

Assertions.assertEquals(Set.of(4, 5), set);

//如果有需要可转为数组
Integer[] intersection = set.toArray(new Integer[0]);

Assertions.assertArrayEquals(new Integer[]{4, 5}, intersection);

2.使用流(Streams)求数组交集

我们可以使用流(Stream) API来遍历一个数组的元素并检查其是否存在于另一个数组中。如果元素存在于第二个数组中，它将被收集到交集结果中。

使用流(Streams)允许我们在同一流管道中对交集元素执行额外的操作，并将结果收集到各种目标类型中。

Integer[] array1 = new Integer[]{1, 2, 3, 4, 5};
Integer[] array2 = new Integer[]{4, 5, 6, 7};

Integer[] intersection = Arrays.stream(array1)
    .distinct()
    .filter(x -> Arrays.asList(array2).contains(x))
    .toArray(Integer[]::new);

Assertions.assertArrayEquals(new Integer[]{4, 5}, intersection);

以上就是Java中计算两个数组的交集的两个常用方法。