350. 两个数组的交集ⅡLeetcode

题目描述

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

示例 1:

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

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

输出结果中每个元素出现的次数，应与元素在两个数组中出现的次数一致。
我们可以不考虑输出结果的顺序。
进阶:

如果给定的数组已经排好序呢？你将如何优化你的算法？
如果 nums1 的大小比 nums2 小很多，哪种方法更优？
如果 nums2 的元素存储在磁盘上，磁盘内存是有限的，并且你不能一次加载所有的元素到内存中，你该怎么办？

思路及解答

/*
思路：
因为题目中要求输出每个元素的次数与在两个数组中出现的次数一致，所以有的元素可能不单单只有一个，所以不可以使用set集合。
这里我们使用的数据结构是map。我们使用数组一中的数作为map的key，每个数出现的次数作为map的值。我们将数组一放入到map当中，
然后遍历数组二，当数组二中的数在数组一中出现一次，那么我们就将map中那个数对应的value值减一，同时将这个数放入一个list集合
当中。如果发现这个数的value值为0，我们就不再添加了。
0，那么我们
*/
class Solution {
    public int[] intersect(int[] nums1, int[] nums2) {
        //新建一个map用来存放nums1中的值
        Map<Integer,Integer> hashMap = new HashMap<Integer,Integer>();
        
        //新建一个list用来存放最终的结果
        List<Integer> res = new ArrayList<Integer>();
        
        //将nums1放入到map集合当中
        for(int i = 0; i < nums1.length; i++){
            Integer value = hashMap.get(nums1[i]);
            hashMap.put(nums1[i],(value == null ? 0 : value) + 1);
        }
        
        //遍历nums2，当nums2中的元素在map中出现时，map中对应的key的value就减一
        for(int i = 0; i < nums2.length; i++){
            if(hashMap.containsKey(nums2[i]) && hashMap.get(nums2[i]) != 0){
                Integer value = hashMap.get(nums2[i]);
                res.add(nums2[i]);
                hashMap.put(nums2[i],value-1);
            }
        }
        
        //将res转化为数组
        int[] result = new int[res.size()];
        Iterator iter = res.iterator();
        int i = 0;
        while(iter.hasNext()){
            result[i++] = (int)iter.next();
        }
        return result;
    }
}