Kotlin中按条件初始化复杂Map的惯用实现方式是什么

最优实现方案 #

首推Kotlin标准库提供的getOrPut扩展函数，这是专为「键不存在时初始化值、存在时直接取值」的场景设计的语法，代码最简洁且完全空安全，不需要手动判空或使用!!断言：

// 一行即可完成判断、初始化、添加元素的全部逻辑
bc.getOrPut(question) { mutableSetOf() }.add(line)

getOrPut的执行逻辑：

• 首先查询question对应的value
• 如果value存在，直接返回该value
• 如果value不存在，执行lambda表达式生成MutableSet实例，存入Map后再返回该实例

原有错误写法的问题说明 #

你尝试的Elvis写法无法运行是两个原因导致的：

1. 语法错误：?: = mutableSetOf(x) 不符合语法规则，Elvis运算符右侧需要是完整表达式，不能直接跟赋值符号
2. 逻辑漏洞：bc[question]?.add(x) 返回的是Boolean?类型，当key存在但add操作返回false（比如Set中已经存在要添加的元素）时，会触发Elvis运算符的右侧逻辑，覆盖原有的Set，导致已有元素丢失。

如果确实要使用Elvis运算符实现，可以参考如下修正后的写法（但可读性和简洁度不如getOrPut）：

// 先确保key对应的Set存在，再执行添加
bc[question] = bc[question] ?: mutableSetOf()
bc[question]?.add(line)

其他可选实现 #

如果你习惯使用JDK集合的原生API，也可以使用computeIfAbsent方法，效果和getOrPut基本一致：

bc.computeIfAbsent(question) { mutableSetOf() }.add(line)

内容的提问来源于stack exchange，提问作者steven smith