Scala：如何从 Set[K] 创建 Map[K,V] 和从 K 到 V 的函数？

Question

从 Set[K] 创建 Map[K,V] 以及从 K 到 V 的函数的最佳方法是什么？

例如，假设我有

scala> val s = Set(2, 3, 5)
s: scala.collection.immutable.Set[Int] = Set(2, 3, 5)

创建

scala> def func(i: Int) = "" + i + i
func: (i: Int)java.lang.String

Map[Int, String](2 -> "22", 3 -> "33", 5 -> "55")Map[Int, String](2 -> "22", 3 -> "33", 5 -> "55")< /代码>

Answer 1

您可以使用 foldLeft：

val func2 = (r: Map[Int,String], i: Int) => r + (i -> func(i))
s.foldLeft(Map.empty[Int,String])(func2)

这会比 Jesper 的解决方案执行得更好，因为 foldLeft 一次性构建了 Map。 Jesper 的代码首先创建一个中间数据结构，然后需要将其转换为最终的Map。

更新：我写了一个微基准测试每个答案的速度：

Jesper (original): 35s 738ms
Jesper (improved): 11s 618ms
           dbyrne: 11s 906ms
         Rex Kerr: 12s 206ms
          Eastsun: 11s 988ms

看起来它们都是只要避免构建中间数据结构，就几乎相同。

Answer 2

怎么样：

(s map { i => i -> func(i) }).toMap

这将 s 的元素映射到元组 (i, func(i))，然后将结果集合转换为 Map 。

注意：i -> func(i) 与 (i, func(i)) 相同。

dbyrne 建议首先创建集合的视图（请参阅他的答案和评论），这可以防止创建中间集合，从而提高性能：

(s.view map { i => i -> func(i) }).toMap

Answer 3

scala> import collection.breakOut
import collection.breakOut

scala> val set = Set(2,3,5)
set: scala.collection.immutable.Set[Int] = Set(2, 3, 5)

scala> def func(i: Int) = ""+i+i
func: (i: Int)java.lang.String

scala> val map: Map[Int,String] = set.map(i => i -> func(i))(breakOut)
map: Map[Int,String] = Map(2 -> 22, 3 -> 33, 5 -> 55)

scala>

Answer 4

除了现有的答案之外，

Map() ++ set.view.map(i => i -> f(i))

它还很短，并且执行得与更快的答案（fold/breakOut）一样好。

（请注意该视图以防止创建新集合；它会在进行过程中进行重新映射。）

Answer 5

其他解决方案缺乏创造力。这是我自己的版本，尽管我真的很想摆脱 _.head 地图。

s groupBy identity mapValues (_.head) mapValues func

Answer 6

与所有伟大的语言一样，有一百万种方法可以完成所有事情。

这是一个将集合与自身压缩在一起的策略。

val s = Set(1,2,3,4,5)
Map(s.zip(s.map(_.toString)).toArray : _*)

编辑： (_.toString) 可以替换为一些返回 V 类型的函数

Answer 7

没有使用“字符串重复”运算符 * 定义 func(i: Int)：

scala> s map { x => x -> x.toString*2 } toMap
res2: scala.collection.immutable.Map[Int,String] = Map(2 -> 22, 3 -> 33, 5 -> 55)