[算法] [leetcode-349] 两个数组的交集

349 两个数组的交集

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

示例 1：

输入：nums1 = [1,2,2,1], nums2 = [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

class Solution {
        public int[] intersection(int[] nums1, int[] nums2) {
            Set<Integer> hashSet = new HashSet<Integer>(2);
            Set<Integer> resultSet = new HashSet<Integer>(2);
            int indexSet = 1;
            if(nums2.length < nums1.length){
                indexSet = 2;
            }
            if(indexSet == 1){
                for(int i=0; i<nums1.length; i++){
                    hashSet.add(nums1[i]);
                }
                for(int j=0; j<nums2.length ; j++){
                    if(hashSet.contains(nums2[j])){
                        resultSet.add(nums2[j]);
                    }
                }
            }else{
                for(int i=0; i<nums2.length; i++){
                    hashSet.add(nums2[i]);
                }
                for(int j=0; j<nums1.length ; j++){
                    if(hashSet.contains(nums1[j])){
                        resultSet.add(nums1[j]);
                    }
                }
            }
            int []resultArray = new int[resultSet.size()];
            int index = 0;
            Iterator<Integer> iterator = resultSet.iterator();
            while(iterator.hasNext()){
                resultArray[index++] = iterator.next();
            }
            return resultArray;
        }
    }