349. 两个数组的交集(java)

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

示例 1:

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

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

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/intersection-of-two-arrays
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

class Solution {
    public int[] intersection(int[] nums1, int[] nums2) {
        Set<Integer> set = new HashSet<>();
        if(nums1 == null||nums2 == null) return null;
        int len1 = nums1.length;
        int len2 = nums2.length;
        for(int i = 0; i < len1; i++){
            set.add(nums1[i]);
        }
        int count = 0;
        for(int i = 0; i < len2; i++){
            if(set.contains(nums2[i])){
                set.remove(nums2[i]);
                nums2[count] = nums2[i];            
                count++;
            }
        }
        return Arrays.copyOfRange(nums2, 0, count);
    }
}