349哈希表-两个数组的交集

题目

 链接：349. 两个数组的交集 - 力扣（LeetCode）

思路

使用set数据结构，先遍历第一个数组，添加到set中，然后遍历第二个数组，看第二个数组中的元素是否在set中含有，有的话加入resSet中。最后申请一个数组来存放resSet结果。

代码

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> set1 = new HashSet<>();
        Set<Integer> resSet = new HashSet<>();
        for(int i : nums1){
            set1.add(i);
        }
        for(int i : nums2){
            if(set1.contains(i)){
                resSet.add(i);
            }
        }
        //申请一个数组存放resSet中的元素，最后返回数组
        int[] arr = new int[resSet.size()];
        int j = 0;
        for(int i : resSet){
            arr[j++] = i;
        }
        return arr;
    }
}