【LeetCode】两个数组的交集 II java

题目

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

示例 1：

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

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

题解

方法1 哈希表

public static int[] intersect(int[] nums1, int[] nums2) {
        int num1_length = nums1.length;
        int num2_length = nums2.length;
        int[] mixLen = num1_length<num2_length?nums1:nums2;
        int[] maxLen = num1_length<num2_length?nums2:nums1;
        Map<Integer, Integer> map = new HashMap<>();
        for (int i = 0; i < mixLen.length; i++) {
            map.put(mixLen[i], map.getOrDefault(mixLen[i],0) + 1);
        }
        int[] ans = new int[mixLen.length];
        int index = 0;
        for (int i = 0; i < maxLen.length; i++) {
            if (map.containsKey(maxLen[i])) {
                ans[index++] = maxLen[i];
                map.replace(maxLen[i], map.get(maxLen[i]) -1);
                if (map.get(maxLen[i]) == 0) {
                    map.remove(maxLen[i]);
                }
            }
        }
        return Arrays.copyOfRange(ans,0,index);
    }

优化

class Solution {
    public int[] intersect(int[] nums1, int[] nums2) {
        // 这里不用判断哪个长度大了
        if (nums1.length > nums2.length) {
            return intersect(nums2, nums1);
        }
        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];
        int index = 0;
        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);
                }
            }
        }
        return Arrays.copyOfRange(intersection, 0, index);
    }
}

方法2 双指针

class Solution {
    public int[] intersect(int[] nums1, int[] nums2) {
        if (nums1.length > nums2.length) {
            return intersect(nums2, nums1);
        }
        Arrays.sort(nums1);
        Arrays.sort(nums2);
        int[] ans = new int[nums1.length];
        int index = 0;
        int i = 0;
        int j = 0;
        while (i<nums1.length && j<nums2.length) {
            if (nums1[i] == nums2[j]) {
                ans[index++] = nums1[i];
                i++;
                j++;
            } else if (nums1[i] > nums2[j]) {
                    j++;
                } else {
                i++;
            }
        }
        return Arrays.copyOfRange(ans,0, index);
    }
}