每日一题:两个数组的交集

地址

难度

简单

题目

给定两个数组 nums1 和 nums2 ,返回 它们的交集 。输出结果中的每个元素一定是 唯一 的。我们可以 不考虑输出结果的顺序 。

示例 1:

输入:nums1 = [1,2,2,1], nums2 = [2,2]

输出:[2]

示例 2:

输入:nums1 = [4,9,5], nums2 = [9,4,9,8,4]

输出:[9,4]

解释:[4,9] 也是可通过的

提示:

  • 1 <= nums1.length, nums2.length <= 1000
  • 0 <= nums1[i], nums2[i] <= 1000

思路

看到数组中有重复的元素,且需要求两个数组的交集,首先就是想到了去重。

在 Java 中可以去重的方式有:使用Set接口,使用Stream API,使用Map,使用Apache Commons库等等。去重之后,就可以简化为 num2 中的数是否与 num1 的数有重复的。

而在 Set 中有一个方法为contains()方法,该方法判断Set中是否包含某个元素,若包含则返回true,否则返回false。

即遍历数组 nums1,对于其中的每个元素,遍历数组 nums2 判断该元素是否在数组 nums2 中,如果存在,则将该元素添加到返回值

class Solution {
    public int[] intersection(int[] nums1, int[] nums2) {
        Set<Integer> set1 = new HashSet<>();
        Set<Integer> set2 = new HashSet<>();
        for (int i : nums1) {
            set1.add(i);
        }

        for (int i : nums2) {
            if (set1.contains(i)) {
                set2.add(i);
            }
        }
        return set2.stream().mapToInt(x->x).toArray();
    }
}