哈希表--两个数组的交集

61 阅读1分钟

题意: 给定两个数组,编写一个函数来计算它们的交集。

image.png

说明: 输出结果中的每个元素一定是唯一的,我们可以不考虑输出结果的顺序。

代码:

public class ArrayIntersect {
    public static int[] arrayIntersect(int[] nums1, int[] nums2) {
        HashSet<Integer> records = new HashSet<>();
        for (int num : nums1) {
            records.add(num);
        }

        HashSet<Integer> result = new HashSet<>();
        for (int num : nums2) {
            if (records.contains(num)) {
                result.add(num);
            }
        }
        return result.stream().mapToInt(o -> o).toArray();
    }

    public static void main(String[] args) {
        int[] nums1 = {1,2,2,1};
        int[] nums2 = {2,2};
        System.out.println(Arrays.toString(arrayIntersect(nums1, nums2)));
    }
}