为何Kotlin中map函数返回Unit列表而非预期的Char列表？

问题出在map函数的lambda表达式的返回逻辑上：

1. 当it == 'a'时，你使用了return@map但未指定返回值，该分支会默认返回Unit类型。
2. Kotlin的类型推断机制会依据lambda中所有分支的返回值确定整体类型：由于存在一个分支返回Unit，编译器会把整个lambda推断为语句体lambda（而非表达式体lambda）。
3. 在语句体lambda中，所有分支的最后一个表达式仅会被当作执行语句，不会作为返回值。因此即便else分支写了it+1，它也不会被返回，整个lambda最终返回的是Unit。
4. 最终map函数会将每个字符都映射为Unit，得到的list自然是Unit类型的列表。

要解决这个问题，需保证两个分支返回相同类型的值：

• 给it == 'a'的分支指定明确返回值（比如返回原字符it），而非无值的return@map；
• 确保lambda所有分支都返回非Unit类型，让编译器将其推断为表达式体lambda。

修改后的示例代码：

fun convert(name:String) {
  val list = name.map {
    if (it == 'a') {
        it // 返回原字符，保持类型一致
    } else { 
        it+1 
    }
  }
  println(list)
}

fun main(args: Array<String>) {
  convert("bbbb") // 输出 [c, c, c, c]
}

内容的提问来源于stack exchange，提问作者kamal MKA