Question

我正试图扭转Kotlin中的地图。到目前为止，我提出了：

mapOf("foo" to 42)
  .toList()
  .map { (k, v) -> v to k }
  .toMap()

有没有更好的方法可以不使用中间人（中间名单）？

Answer 1

由于Map由Entry组成且不是Iterable。但是您可以使用Map#entries代替它，它将映射到Map#entrySet以创建Set<Entry>的支持视图，例如：

val reversed = map.entries.associateBy({ it.value }) { it.key }

或使用Iterable#associate，但会创建额外的Pair。

val reversed = map.entries.associate{(k,v)-> v to k}

或使用Map#forEach代替。例如：

val reversed = mutableMapOf<Int, String>().also {
    //     v-- use `forEach` here     
    map.forEach { (k, v) -> it.put(v, k) } 
}.toMap()
// ^--- you can add `toMap()` to create an immutable Map.

Answer 2

这是一个简单的扩展函数，它可以反转地图 - 不会产生不必要的垃圾（如对，中间数据结构和不必要的闭包）

fun <K, V> Map<K, V>.reversed() = HashMap<V, K>().also { newMap ->
    entries.forEach { newMap.put(it.value, it.key) }
}

请注意，apply是内联的，entries.forEach也是内联的（Map::forEach不一样）

Answer 3

我还在学习Kotlin的细节，但我有同样的要求，而Kotlin 1.2看起来你可以迭代一个Map，所以就像这样直接映射（）：

@Test
fun testThatReverseIsInverseOfMap() {
    val intMap = mapOf(1 to "one", 2 to "two", 3 to "three")
    val revMap = intMap.map{(k,v) -> v to k}.toMap()
    assertTrue(intMap.keys.toTypedArray() contentEquals revMap.values.toTypedArray())
    assertTrue(intMap.values.toTypedArray() contentEquals revMap.keys.toTypedArray())
}

Answer 4

如果您的地图不是1-1映射，并且您希望倒置为值列表：

mapOf(1 to "AAA", 2 to "BBB", 3 to "BBB").toList()
        .groupBy { pair -> pair.second } // Pair<Int, String>
        .mapValues { entry -> 
          entry.value.map { it.first } // Entry<String, List<Pair<Int, String>>
        }

Answer 5

如果您需要将m: Map<K, List<V>>之类的多图逆转为Map<V, List<K>>，则可以

m
  .flatMap { it.value.map { oneValue -> oneValue to it.key } }
  .groupBy({ it.first }, { it.second })
  .toMap()

依次

mapOf('a' to listOf('b', 'c'), 'd' to listOf('b')) 平面映射到类似
listOf('b' to 'a', 'c' to 'a', 'b' to 'd')归为
listOf('b' to listOf('a', 'd'), 'c' to listOf('a'))，然后将其转换为地图。

这可能会创建中间对象。

如何在Kotlin中反转地图？

5 个答案: