两个数组的交集II

力扣 350. 两个数组的交集 II

题目说明

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

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

示例

示例 1:

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

示例 2:

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

笔者理解

此题是一道数组算法问题，在力扣题库中被定义为简单题。

解法

当笔者阅读完此题后，直接用哈希表进行统计，简单粗暴，让我们来看看具体如何实现的吧。

实现

class Solution {
    /*
     * 哈希表
     * 数组
     */
    public int[] intersect(int[] nums1, int[] nums2) {
        // 保证数组相对大小，简化操作
        if (nums1.length > nums2.length) {
            return intersect(nums2, nums1);
        }
        int size1 = nums1.length;
        int size2 = nums2.length;

        // 分别统计两数组各数组出现的次数
        HashMap<Integer, Integer> map1 = new HashMap<>();
        HashMap<Integer, Integer> map2 = new HashMap<>();

        for (int i = 0; i < size2; i++) {
            if (i < size1) {
                map1.put(nums1[i], map1.getOrDefault(nums1[i], 0) + 1);
            }
            map2.put(nums2[i], map2.getOrDefault(nums2[i], 0) + 1);
        }

        List<Integer> list = new ArrayList<>();

        int key, value;
        // 遍历取得两个数组里面都出现过的数
        for (Map.Entry<Integer, Integer> entry : map1.entrySet()) {
            key = entry.getKey();
            value = entry.getValue();
            
            if (value > 0 && map2.getOrDefault(key, 0) > 0) {
                int times = Math.min(value, map2.get(key));
                while (times-- > 0) {
                    list.add(key);
                }
            }
        }

        int[] result = new int [list.size()];

        for (int i = 0; i < list.size(); i++) {
            result[i] = list.get(i);
        }

        return result;
    }
}