【LeeCode】350. 两个数组的交集 II

【题目描述】

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

​​https://leetcode.cn/problems/intersection-of-two-arrays-ii/​​

【示例】

【代码】​​leecode​​

package com.company;
// 2023-03-09
import java.util.*;

class Solution {
    public int[] intersect(int[] nums1, int[] nums2) {
        Arrays.sort(nums1);
        Arrays.sort(nums2);
        int len = nums1.length;
        int len2 = nums2.length;
        if (len > len2) return intersect(nums2, nums1);

        Map<Integer, Integer> map = new HashMap<>();
        for (int x : nums1){
            map.put(x, map.getOrDefault(x, 0) + 1);
        }
        int index = 0;
        int[] res = new int[len];
        for (int num : nums2){
            int count = map.getOrDefault(num, 0);
            // 如果大于0，表示数组1中有这个值
            if (count > 0){
                // 赋值给res
                res[index++] = num;
                // 原数量减一
                count--;
                // 如果还有库存, 则更新count
                if (count > 0){
                    map.put(num, count);
                }else {
                    // 如果库存为0, 则需要移除
                    map.remove(num);
                }
            }
        }
        return Arrays.copyOfRange(res, 0, index);
    }
}

public class Test {
    public static void main(String[] args) {
        new Solution().intersect(new int[]{1,2,2,1}, new int[]{2});   // 输出：[2]
        new Solution().intersect(new int[]{1,2,2,1}, new int[]{2, 2});   // 输出：[2,2]
        new Solution().intersect(new int[]{4,9,5}, new int[]{9,4,9,8,4});   // 输出：[4,9]
    }
}

【代码】​​leecode​​

package com.company;
// 2023-03-09
import java.util.*;

class Solution {
    public int[] intersect(int[] nums1, int[] nums2) {
        Arrays.sort(nums1);
        Arrays.sort(nums2);
        int len = nums1.length;
        int len2 = nums2.length;

        int[] res = new int[Math.min(len, len2)];
        int index1 = 0;
        int index2 = 0;
        int index = 0;

        while (index1 < len && index2 < len2){
            if (nums1[index1] < nums2[index2]){
                index1++;
            }else if (nums1[index1] > nums2[index2]){
                index2++;
            }else {
                res[index] = nums1[index1];
                index1++;
                index2++;
                index++;
            }
        }
        return Arrays.copyOfRange(res, 0, index);
    }
}

public class Test {
    public static void main(String[] args) {
        new Solution().intersect(new int[]{1,2,2,1}, new int[]{2});   // 输出：[2]
        new Solution().intersect(new int[]{1,2,2,1}, new int[]{2, 2});   // 输出：[2,2]
        new Solution().intersect(new int[]{4,9,5}, new int[]{9,4,9,8,4});   // 输出：[4,9]
    }
}