Map的computeIfAbsent的使用场景和使用方法

最新推荐文章于 2025-10-22 16:41:26 发布

原创最新推荐文章于 2025-10-22 16:41:26 发布 · 4.4k 阅读

6 ·

CC 4.0 BY-SA版权

文章标签：

#java #hashmap

本文介绍了如何使用Java中的Map接口提供的computeIfAbsent方法来简化复杂映射操作，特别是在需要确保某个键对应的值是一个集合，并在集合中添加元素的场景下。通过对比传统的if条件判断和put操作，展示了computeIfAbsent方法如何使代码更加简洁和高效。示例中，将一个整数数组中的重复数字归并到对应的List中，computeIfAbsent确保了每个数字只创建一个List，并自动将相同数字添加到对应的List中。

Map的computeIfAbsent的使用场景和使用方法

default V computeIfAbsent(K key, Function<? super K, ? extends V> mappingFunction) {
    ...
}

我们在复杂map操作（put操作）时候有的时候不知道此时当前key对应的value值是否存在，这里，我们如果使用常规的代码编写，代码量比较大

例如我们定义一个场景：存在一个数组，我们需要将当前数组中相同的数存储到一个对应List集合中

常规实现：

    @Test
    public void test02() {
        //复杂map的使用场景：首先我们说复杂map，即map的value值为一个list集合或者是一个Set集合，对象或者是其他的集合
        //给定一个场景：现在存在一个数组，我们需要将当前数组中相同的数存储到一个对应List集合中
        int[] nums = {1,2,3,1,3,4,6,7,9,9,1,3,4,5};
        Map<Integer, List<Integer>> map = new HashMap<>();
        //普通的写法
        for (int i = 0; i < nums.length; i++) {
            if (!map.containsKey(nums[i])) {
                ArrayList<Integer> list = new ArrayList<>();
                list.add(nums[i]);
                map.put(nums[i],list);
            }else {
                map.get(nums[i]).add(nums[i]);
            }
        }
        map.forEach((key,value) -> {
            System.out.print(key + ": ");
            System.out.println(value);
        });
    }

使用computeIfAbsent方法实现：

    @Test
    public void  test03() {
        int[] nums = {1,2,3,1,3,4,6,7,9,9,1,3,4,5};
        Map<Integer, List<Integer>> map = new HashMap<>();
        //我们使用map的computeIfAbsent解决
        for (int i = 0; i < nums.length; i++) {
            //返回值是该key对应的集合list
            map.computeIfAbsent(nums[i], key -> new ArrayList<Integer>());
            map.get(nums[i]).add(nums[i]);
        }
    
        map.forEach((key,value) -> {
            System.out.print(key + ": ");
            System.out.println(value);
        });
    }

输出结果：

1: [1, 1, 1]
2: [2]
3: [3, 3, 3]
4: [4, 4]
5: [5]
6: [6]
7: [7]
9: [9, 9]