leetcode-349. 两个数组的交集

class Solution {
    public int[] intersection(int[] nums1, int[] nums2) {
        // 用于保存 nums1 的 HashSet
        Set<Integer> set1 = new HashSet<>();
        // 用于存储相交元素的 HashSet
        Set<Integer> set2 = new HashSet<>();
        // 遍历 nums1 将其存到 set1 中
        for(int i:nums1){
            set1.add(i);
        }
        // 若 nums2 中存在 set1 中的元素，则加入 set2
        for(int i:nums2){
            if(set1.contains(i)){
                set2.add(i);
            }
        }
        // 开辟一个长度为 set2.size() 的存放相交元素的数组
        int[] arr = new int[set2.size()];
        int j = 0;
        // 把 set2 里的值存到 arr 数组中
        for(int i:set2){
            arr[j++] = i;
        }
        return arr;
    }
}