LeetCode算法题-Intersection of Two Arrays II（Java实现）_intersection of two arrays ii java-优快云博客

这是悦乐书的第208次更新，第220篇原创

01 看题和准备

今天介绍的是LeetCode算法题中Easy级别的第76题（顺位题号是350）。给定两个数组，编写一个函数来计算它们的交集。例如：

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

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

注意：

结果中的每个元素应该出现在两个数组中显示的次数。
结果可以是任何顺序。

跟进：

如果给定的数组已经排序怎么办？你会如何优化你的算法？
如果nums1的尺寸与nums2的尺寸相比较小怎么办？哪种算法更好？
如果nums2的元素存储在磁盘上，并且内存有限，以致您无法一次将所有元素加载到内存中，该怎么办？

02 第一种解法

今天这题是昨天题目的升级版，唯一的区别就是最后输出的结果数组中，元素并不唯一。对此，只需要将昨天的HashSet换成ArrayList即可，双指针的思路还是一样。

先将两数组排序，然后使用双指针，依次判断两数组中的元素是否相等，如果某个元素大于或小于另外一个元素，则将指针向后移动，如果相等，则将元素放入ArrayList中，然后将ArrayList中的元素迭代放入数组，最后返回。

因为使用Arrays类的sort方法，所以时间复杂度是O(n log(n))，空间复杂度是O(n)。

public int[] intersect(int[] nums1, int[] nums2) {
    List<Integer> list = new ArrayList<Integer>();
    Arrays.sort(nums1);
    Arrays.sort(nums2);
    int i = 0;
    int j = 0;
    while (i < nums1.length && j < nums2.length) {
        if (nums1[i] < nums2[j]) {
            i++;
        } else if (nums1[i] > nums2[j]) {
            j++;
        } else {
            list.add(nums1[i]);
            i++;
            j++;
        }
    }
    int[] result = new int[list.size()];
    int k = 0;
    for (Integer num : list) {
        result[k++] = num;
    }
    return result;
}

03 第二种解法

利用HashMap，先将其中一个数组的元素全部存入其中，key为元素值，value为该元素出现次数。然后遍历第二个数组，如果map中存在当前元素，并且在map中此元素所对应的value值大于0，就将其添加进ArrayList中，同时对该元素在map中所对应的value值减1，避免重复判断。

此解法因为用到了HashMap的contains方法，因此时间复杂度最好情况是O(n)，最坏情况是O(n^2)，空间复杂度是O(n)。

public int[] intersect2(int[] nums1, int[] nums2) {
    Map<Integer, Integer> map = new HashMap<Integer, Integer>();
    for (int num : nums1) {
        map.put(num, map.getOrDefault(num, 0) + 1);
    }
    List<Integer> result = new ArrayList<Integer>();
    for (int num : nums2) {
        if (map.containsKey(num) && map.get(num) > 0) {
            result.add(num);
            map.put(num, map.get(num) - 1);
        }
    }
    int[] arr = new int[result.size()];
    for (int i = 0; i < result.size(); i++) {
        arr[i] = result.get(i);
    }
    return arr;
}