kotlin flatten 与 flatMap

胡小牧2024-01-25 21:47

kotln中 flatten 和 flatMap 在 Kotlin 中虽然都用于扁平化处理集合，但它们的用法和效果并不完全一样

flatten

flatten 函数主要应用于嵌套集合（如 List of List 或 Set of Set 等），它会将嵌套集合中的所有元素合并到一个单一层次的集合中。

复制代码

val nestedList = listOf(listOf("a", "b"), listOf("c", "d"))
val flattenedList = nestedList.flatten() // ["a", "b", "c", "d"]

flatMap

flatMap 不仅可以将嵌套集合扁平化，更重要的是它允许你对每个子集合应用一个映射函数来生成新的元素。这个映射函数的结果应当是一个集合，然后这些结果集会被合并成一个新的集合

复制代码

val lists = listOf(listOf("hello"), listOf("world"))
val chars = lists.flatMap { it.map { it.toCharArray() } } // ['h', 'e', 'l', 'l', 'o', 'w', 'o', 'r', 'l', 'd']