leetcode305 两个数组的交集Ⅱ

给你两个整数数组 nums1 和 nums2 ，请你以数组形式返回两数组的交集。返回结果中每个元素出现的次数，应与元素在两个数组中都出现的次数一致（如果出现次数不一致，则考虑取较小值）。可以不考虑输出结果的顺序。

class Solution {
    public int[] intersect(int[] nums1, int[] nums2) {
        //如果nums1元素个数大于nums2元素个数，交换数组位置
        if (nums1.length > nums2.length) {
            return intersect(nums2, nums1);
        }
        //将nums1中的每一个元素都添加到hashmap中，对已存在的元素进行相对应的计数
        Map<Integer, Integer> map = new HashMap<Integer, Integer>();
        for (int num : nums1) {
            int count = map.getOrDefault(num, 0) + 1;
            map.put(num, count);
        }
        int[] intersection = new int[nums1.length];
        //定义一个变量index，用来记录当前交集里的数字个数
        int index = 0;
        //遍历nums2，如果当前元素存在，且计数为正，将其放入到intersection数组中
        for (int num : nums2) {
            int count = map.getOrDefault(num, 0);
            if (count > 0) {
                intersection[index++] = num;
                count--;
                if (count > 0) {
                    map.put(num, count);
                } else {
                    map.remove(num);
                }
            }
        }
        //以数组方式返回intersection的前0-index个数字
        return Arrays.copyOfRange(intersection, 0, index);
    }
}

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