题目

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

示例 1：

输入：nums1 = [1,2,2,1], nums2 = [2,2] 输出：[2,2]

示例 2：

输入：nums1 = [4,9,5], nums2 = [9,4,9,8,4] 输出：[4,9]

说明：

• 输出结果中每个元素出现的次数，应与元素在两个数组中出现次数的最小值一致。
• 我们可以不考虑输出结果的顺序。

思路

我们可以把其中一个数组构建成一个哈希表，然后通过遍历另一个数组找到交集，并将交集元素存入一个结果数组中返回。

Map<Integer,Integer> map = new HashMap<>();//哈希表
int[] result = new int[nums1.length];//结果数组

细节一

• 时间复杂度是 O(m + n) ，其中 m 和 n 分别为两个数组的长度，耗时主要在建表和遍历上。
• 而空间主要耗在哈希表上，所以我们要考虑将长度较短的数组拿来建表，可得空间复杂度为 O(min(m,n)) 。
• 为了方便编码，我们规定 nums1 始终是元素较少的那个数组，如果不符合我们必须考虑首先将两数组位置进行交换。

		//保证nums1永远都是元素少的那个数组
        if (nums1.length > nums2.length) {
    
    
            return intersect(nums2, nums1);
        }

细节二

• 我们并不清楚交集元素的个数，但显然它必定不会超过 nums1 的元素个数，所以我们将 nums1 的长度作为结果数组的长度。
• 在最后我们需要考虑截断结果数组再 return。

		//截断至结果数组最后一个有效数据并返回
        return Arrays.copyOfRange(result,0,index);

方法介绍

Arrays.copyOfRange(T[] original,int from,int to)

将一个原始的数组 original ，从下标 from 开始复制，复制到下标 to ，生成一个新的数组。
注意这里包括下标 from ，但不包括下标 to 。

代码

class Solution {
    
    
    public int[] intersect(int[] nums1, int[] nums2) {
    
    
        //保证nums1永远都是元素少的那个数组
        if (nums1.length > nums2.length) {
    
    
            return intersect(nums2, nums1);
        }
        Map<Integer,Integer> map = new HashMap<>();
        int[] result = new int[nums1.length];//结果数组（可以肯定其大小肯定小于或等于nums1的大小）
        int index = 0;//结果数组的下标
        //使用哈希表记录nums1的元素类型和元素个数
        for (int i = 0; i < nums1.length; i++) {
    
    
            if (!map.containsKey(nums1[i])) {
    
    
                map.put(nums1[i],1);
            } else {
    
    
                map.replace(nums1[i],map.get(nums1[i]) + 1);
            }
        }
        //通过查表和遍历nums2寻找交集
        for (int i = 0; i < nums2.length; i++) {
    
    
            if (map.containsKey(nums2[i]) && map.get(nums2[i]) > 0) {
    
    
                result[index] = nums2[i];
                map.replace(nums2[i],map.get(nums2[i]) - 1);
                index++;
            }
        }
        //截断至结果数组最后一个有效数据并返回
        return Arrays.copyOfRange(result,0,index);
    }
}

提交结果