合并2个地图与键作为第二个地图的值

Question

我有两张地图被定义为Map

map1 : {
   Honda=[Accord, Civic, CRV], 
   Toyota=[Corola, RAV4, Yaris]
}
map2 : {
   Accord=[5w30, 10w40],
   Civic=[0w30, 5w30, 5w40],
   CRV=[5w30, 5w40, 10w40],
   Corola=[0w30, 10w40], 
   RAV4=[0w20, 0w30, 5w30]
}

我需要合并这两个地图，这样结果地图键将是品牌名称（如本田、丰田），值将是所有的油。有些是这样的：

mergeMap: {
   Honda=[0w30, 5w30, 5w40, 10w40],
   Toyota=[0w20, 0w30, 5w30, 10w40]
}

我编写了sam old school method，它使用forEach块，但我认为使用流等更好。可能吗？如果是，怎么做？




Answer 1

由于生成的贴图将具有相同的键，因此可以使用mapValues。

在lambda中，flatMap将值中的每个字符串设置为第二个映射中的相应值（如果存在）。

map1.mapValues { (_, v) -> v.flatMap { map2[it] ?: emptyList() }.toSortedSet() }

这给你一个地图

请注意，这将按字典顺序对“0w30”字符串进行排序，因为它们是字符串，所以你不会得到你想要的确切顺序。我不知道这些字符串代表什么，但是你可能应该用一个实现可比较的数据类来表示它们。这会给你正确的顺序。