Leetcode350.两个数组的交集II

---

前言

隐约记得做过这道题，但就是有点懵。。

一、题目描述

直接放图就完事了

二、大致思路

线利用哈希表存储较短数组nums1的元素及其出现次数，然后遍历另一个数组nums2，对于其每一个元素a：
若a出现在nums1中，则将a添加到答案数组中，同时a在nums1中的出现次数减1。若减1后出现次数仍大于0，则改变哈希表中a的出现次数（hashmap.put()）。

具体代码如下：

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[] ans = new int[nums1.length];
        int index = 0;
        for(int n:nums2){
            int count = map.getOrDefault(n, 0);
            if(count > 0){
                ans[index++] = n;
                count--;
                if(count > 0){
                    map.put(n, count);
                }else{
                    map.remove(n);
                }
            }
        }
        return Arrays.copyOfRange(ans, 0, index);
    }
}

总结

学习了hashmap的getOrDefault()方法以及Arrays.copyOfRange()方法。