按键列表 (Kotlin) 分组



假设我有一个

class Question(val tags:List<String>, val text:String)

除了其他属性外,它(显然(还有多个标签。

我想将许多问题实例转换为(单个!(标签到问题映射,例如:Map<String,List<Question>>.

我该怎么做?一个简单的groupBy { it.tags }提供了一个Map<List<String>,List<Question>>

一个通用的扩展函数,因为我已经需要它几次了:

fun <T, K> Iterable<T>.groupByMany(
keyExtractor: (T) -> Iterable<K>
): Map<K, List<T>> = mutableMapOf<K, MutableList<T>>()
.also { grouping ->
forEach { item ->
keyExtractor(item).forEach { key ->
grouping.computeIfAbsent(key) { mutableListOf() }.add(item)
}
}
}

用法:

val byTag = questions.groupByMany { it.tags }
val questions: Iterable<Question> = ....
val map = HashMap<String, MutableList<Question>>()
questions.forEach { question ->
question.tags.forEach { tag ->
val otherQuestions = map[tag]
if (otherQuestions == null) map[tag] = arrayListOf(question)
else otherQuestions.add(question)
}
}
val resultMap: Map<String, List<Question>> = map

最新更新