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

Posted 2023-03-10 孤注一掷 、

题目

 链接：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;