349. 两个数组的交集 - 力扣（LeetCode） (leetcode-cn.com)

题目

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

实例：

输入：nums1 = [1,2,2,1], nums2 = [2,2]
输出：[2]

解题思路

• 利用set不能存储相同值得原理，将两个数组转为set。
• 在使用retainAll()方法，移除掉set1中不包含set2的元素，则取出他们的交集

代码

public int[] intersection(int[] nums1, int[] nums2) {
        //定义set集合
        Set<Integer> set1 = new HashSet<>();
        Set<Integer> set2 = new HashSet<>();
        //写值
        for (int num : nums1) {
            set1.add(num);
        }
        for (int num : nums2) {
            set2.add(num);
        }

        set1.retainAll(set2);//就是移除 set1 中的所有的不包含 set2 中的元素
        //因为set是没有重复元素的，所有吧set转为数组输出即可
        int i = 0;
        int[] arr = new int[set1.size()];
        for (int item : set1) {
            arr[i] = item;
            i++;
        }
        return arr;
    }