题目链接

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

示例 1:

  1. 输入:nums1 = [1,2,2,1], nums2 = [2,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集合类型,不允许重复值。HashSet重复值会覆盖,元素唯一。因此,将数组1转成无重复元素的HashSet,再判断数组2中的元素是否包含在数组1中。

class Solution {
    public int[] intersection(int[] nums1, int[] nums2) {
        if(nums1 == null || nums1.length == 0 || nums2 == null || nums2.length==0){
            return new int[0];
        }
        Set<Integer> nums1Set = new HashSet<>(); // nums1数组的Set
        Set<Integer> resultSet = new HashSet<>(); // 结果Set
        // 遍历nums1数组
        for (int i : nums1) {
            nums1Set.add(i);
        }
        // 遍历nums2数组并写入结果 {
        for (int i : nums2) {
            if(nums1Set.contains(i)) {
                resultSet.add(i);
            }
        }
        int[] result = new int[resultSet.size()];
        int index = 0;
        // 集合转结果数组
        for (Integer integer : resultSet) {
            result[index] = integer;
            index++;
        }
        return result;
    }
}