【哈希表】349.两个数组的交集（简单）

349.两个数组的交集（简单）

题目

力扣题目链接

给定两个数组 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

思路一：使用HashSet

将nums1的元素一次add进set集合中，可以保证重复的元素在set集合中只出现一次，之后循环nums2的元素，判断如果当前元素在set集合中存在，则代表有交集，将当前元素add进retSet即可， 至于retSet，由于nums2中的元素很可能也存在重复，所以也要使用set存储结果集。

代码

public int[] intersection(int[] nums1, int[] nums2) {
    Set<Integer> set = new HashSet<>();
    Set<Integer> retSet = new HashSet<>();
    for (int j : nums1) {
        set.add(j);
    }
    for (int j : nums2) {
        if (set.contains(j)) {
            retSet.add(j);
        }
    }
    return retSet.stream().mapToInt(x -> x).toArray();
}

时间复杂度

O(n)

思路二：使用HashMap

将两个数组分别put进两个hashmap，因为hashmap天生key不存在重复，之后再循环第一个map，判断每个键在map2中是否存在，如果不存在则remove，存在则不需要操作，最后将map的key转为数组即可。

代码

public int[] intersection02(int[] nums1, int[] nums2) {
    Map<Integer, Integer> map = new HashMap<>();
    Map<Integer, Integer> map2 = new HashMap<>();
    for (int n : nums1) {
        map.put(n, -1);
    }
    for (int n : nums2) {
        map2.put(n, -1);
    }
    Iterator<Map.Entry<Integer, Integer>> iterator = map.entrySet().iterator();
    while (iterator.hasNext()) {
        Map.Entry<Integer, Integer> next = iterator.next();
        if(map2.get(next.getKey()) == null) {
            iterator.remove();
        }
    }
    return map.keySet().stream().mapToInt(x -> x).toArray();
}

简化写法

public int[] intersection02(int[] nums1, int[] nums2) {
    Map<Integer, Integer> map = new HashMap<>();
    Map<Integer, Integer> map2 = new HashMap<>();
    for (int n : nums1) {
        map.put(n, -1);
    }
    for (int n : nums2) {
        map2.put(n, -1);
    }
    map.entrySet().removeIf(next -> map2.get(next.getKey()) == null);
    return map.keySet().stream().mapToInt(x -> x).toArray();
}

时间复杂度

O(n)