Map.computeIfAbsent(...)函数的使用

Map.computeIfAbsent(…)函数的使用

前言

最近遇见一个问题：有一批学生，请把这批学生按照班级分组。
脑子里第一个想法使用Map<String, List<Student>>数据结构，刚好这就让我学会了一个新的函数Map.computeIfAbsent(...)

1、函数的定义

computeIfAbsent官方文档说明
大致的意思就是：如果指定的键尚未与某个值关联，则尝试使用给定的映射函数计算其值，并将该值输入此映射，除非该值为空。如果函数返回空值，则不记录映射。
最常见的用法就是构造一个新对象作为初始值computeIfAbsent(key, k->new Value(f (k) ) )
或者要实现多值映射 Map<K, Collection(V)> 支持每个键有多个值map.computeIfAbsent(key, k->new HashSet<V>.add(V))

2、问题解决

//学生类
public class Student {
    private String classNo;
    private int age;
    private String sex;
    ...
}

public static void main(String[] args) {
	List<Student> students = new ArrayList<>();
	Student stu = new Student();
	stu.setClassNo("1001");
	stu.setAge(21);
	stu.setSex("男生");
	students.add(stu);
	//初始化List
	...
	students.add(stu);
	/**
	* students值为
	* [
	* Student {classNo = '1001', age = 21, sex = '男生'},
	* Student {classNo = '1002', age = 20, sex = '女生'},
	* Student {classNo = '1003', age = 21, sex = '男生'},
	* Student {classNo = '1001', age = 20, sex = '女生'},
	* Student {classNo = '1002', age = 21, sex = '女生'},
	* Student {classNo = '1003', age = 20, sex = '男生'},
	* ]
	**/
	
	/**
	* 情况1：判断以当前班级编号为key的map是否存在对应的value
	* 存在的话就把当前学生添加到对应的value中
	* 不存在就先添加映射，再添加当前学生到对应的value中
	**/
	Map<String, List<Student>> map = new HashMap<>();
	for(Student item : students){
		List<Student> list = map.get(item.getClassNo);
		if(CollectionUtils.isEmpty(list)){
			list = new ArrayList<>();
			map.put(item.getClassNo, list);
		}
		list.add(item);
	}
	
	/**
	* 情况2：直接使用complateIfAbsent()函数
	**/
	Map<String, List<Student>> map2 = new HashMap<>();
	for(Student item : students){
		map2.complateIfAbsent(item.getClassNo, k->new ArrayList<>()).add(item);
	}
}

3、总结

很容易看出来，情况2代码极其简洁，是不是学到了新知识
接下来，我来解释一下在内存中是怎么表现的：

1、map中存在1001的key情况下，会在右边的list中添加一个当前学生

2、map中不存在1002的key,会先添加一个1002为key的初始话list，然后再在list中添加一个当前学生

结语：粗糙理解，分享给大家，如有疑问、错误欢迎指出！