leetcode349. 两个数组的交集

给定两个数组，编写一个函数来计算它们的交集。

示例 1:

输入: nums1 = [1,2,2,1], nums2 = [2,2]
输出: [2]
示例 2:

输入: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出: [9,4]
说明:

输出结果中的每个元素一定是唯一的。
我们可以不考虑输出结果的顺序。

思路：

把一个数组的数字放set里，遍历另一个数组判断在不在set里即可。

class Solution {
    public int[] intersection(int[] nums1, int[] nums2) {
        Set<Integer> set = new HashSet<>();
        int minLen = Math.min(nums1.length, nums2.length);
        int[] res = new int[minLen];
        
        for(int num : nums1){
            set.add(num);
        }
        
        int k = 0;
        for(int num : nums2){
            if(set.contains(num)){
                res[k++] = num;
                set.remove(num);
            }    
        }
        
        return Arrays.copyOf(res, k);
    }
}