Map.getOrDefault()方法 以解两个数组的交集为例

Map.getOrDefault()

格式

map.getOrDefault(Object key, V defaultValue) 

第一个值为要查询的key值，第二个值为没有对应key时返回的默认值。
换句话说，当集合中有这个key时，就使用这个key对应的value值，如果没有就使用最开始设置的默认值，也就是defaultValue

示例

int count = map.getOrDefault(y, 0);

查询map中key为y的value，如果没有key为y的数据，则默认返回0

实际使用案例

350. 两个数组的交集 II

题目

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

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

代码

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 x : nums1) {
            //getOrDefault() 方法获取指定 key 对应对 value，如果找不到 key ，则返回设置的默认值。
            //在这里就相当于，获取key为x的值的value，如果没有，就默认为1。后面再次加一是为了统计某个数在数组中出现的次数
            int count = map.getOrDefault(x, 0) + 1;
            map.put(x, count);
        }
        int[] intersection = new int[nums1.length];
        int index = 0;
        for (int y : nums2) {
            int count = map.getOrDefault(y, 0);
            if (count > 0) {
                intersection[index++] = y;
                count--;
                if (count > 0) {
                    map.put(y, count);
                } else {
                    map.remove(y);
                }
            }
        }
        return Arrays.copyOfRange(intersection, 0, index);
    }
}