Kotlin：将对象列表转换为对象属性映射

Question

使用以下数据类：

data class Photo(val id: String)
data class Album(val id: String, val photos: List<Photo>)

我希望能够将Albums 的列表转换为Map<photoId, Set<albumId>> 的地图

例如：

val albums = listOf(
    Album("1", listOf(
        Photo("1"),
        Photo("2"),
        Photo("3")
    )),
    Album("2", listOf(
        Photo("1"),
        Photo("2")
    ))
)

会变成：

val map = mapOf(
    Pair("1", setOf("1", "2")),
    Pair("2", setOf("1", "2")),
    Pair("3", setOf("1")
)

有没有办法使用 kotlin 转换函数/lambdas 来做到这一点？例如，groupBy、flatMap？

Answer 1

更清洁的解决方案

val grouped = albums
    .flatMap { album -> album.photos.map { photo -> Pair(photo.id, album.id) } }
    .groupBy( { (photoId, _) -> photoId }, { (_, albumId) -> albumId } )
    .mapValues { it.value.toSet() }

并检查正确性

val map = mapOf(
    Pair("1", setOf("1", "2")),
    Pair("2", setOf("1", "2")),
    Pair("3", setOf("1"))
)

check(map == grouped) // Success

一点解释

• 使用第一个 flatMap，我们将 专辑 列表 (List<Album>) 转换为 List<Pair<String, String>>（photo.id 到 album.id）。

  李>

• 使用groupBy，我们将第二个列表转换为Map<String, List<String>>，其中键为photo.id，并按album..id分组

• 使用mapValues，我们将地图转换为Map<String, Set<String>>

Answer 2

一个肮脏的解决方案：

val map = albums.map {
    it.photos.map { it.id } // get all photos in all album
}.flatten() // into a single list (instead of 2-d list)
    .toSet() // remove duplicates
    .associate { photo ->
        photo to albums.filter { it.photos.contains(Photo(photo)) } // for each photo, find its existance in all albums
            .map { it.id }.toSet() // output album ids
    }