java 选取两个数组相同元素

原创

mob649e816594b7 2024-08-15 06:49:12 ©著作权

文章标签 数组 java Java 文章分类 Java 后端开发

©著作权归作者所有：来自51CTO博客作者mob649e816594b7的原创作品，请联系作者获取转载授权，否则将追究法律责任

Java中选取两个数组相同元素的方法

在我们日常的编程工作中，常常需要比较两个数组并找出它们的共同元素。Java语言为我们提供了多种方法来实现这一功能。在本文中，我们将探讨一种简单而有效的方法来选取两个数组的相同元素，并辅以代码示例和序列图以帮助理解。

方法概述

我们可以使用集合（Set）数据结构来找出两个数组的交集。下面是我们将采取的步骤：

创建两个数组。
利用Java的集合框架，将其中一个数组转化为Set。
遍历另一个数组，检查每个元素是否在这个Set中出现。
如果元素存在，则将其添加到结果列表中。

代码示例

以下是完整的Java示例代码，展示了如何实现上述步骤：

import java.util.Arrays;
import java.util.HashSet;
import java.util.Set;

public class ArrayIntersection {
    public static void main(String[] args) {
        // 定义两个数组
        int[] array1 = {1, 2, 3, 4, 5};
        int[] array2 = {4, 5, 6, 7, 8};
        
        // 调用方法获取交集
        int[] intersection = getIntersection(array1, array2);
        
        // 打印结果
        System.out.println("相同元素: " + Arrays.toString(intersection));
    }

    public static int[] getIntersection(int[] array1, int[] array2) {
        Set<Integer> set = new HashSet<>();
        
        // 将第一个数组的元素添加到Set中
        for (int num : array1) {
            set.add(num);
        }

        // 用集合来存储相同元素
        Set<Integer> intersectionSet = new HashSet<>();
        
        // 遍历第二个数组，查找相同元素
        for (int num : array2) {
            if (set.contains(num)) {
                intersectionSet.add(num);
            }
        }

        // 将Set转化为数组
        int[] intersection = new int[intersectionSet.size()];
        int i = 0;
        for (int num : intersectionSet) {
            intersection[i++] = num;
        }
        
        return intersection;
    }
}

代码解释

上面的代码中，我们定义了两个整数数组 array1 和 array2。我们利用一个 HashSet 来存储 array1 的元素，并遍历 array2，查找其元素是否在 HashSet 中。如果找到相同的元素，就将它添加到 intersectionSet 中。最后，我们将这个集合转换为数组并返回。

序列图

为了更好地理解程序的执行过程，下面是展示数组相同元素选取过程的序列图：

sequenceDiagram
    participant User
    participant Main
    participant ArrayIntersection

    User->>Main: 定义 array1, array2
    Main->>ArrayIntersection: getIntersection(array1, array2)
    ArrayIntersection->>ArrayIntersection: 初始化 set
    ArrayIntersection->>ArrayIntersection: 填充 set
    ArrayIntersection->>ArrayIntersection: 遍历 array2
    ArrayIntersection->>set: check contains(num)
    alt 元素存在
        ArrayIntersection->>intersectionSet: add(num)
    else 元素不存在
        ArrayIntersection-->>Main: 继续遍历
    end
    ArrayIntersection-->>Main: 返回相同元素数组
    Main-->>User: 打印相同元素