通过比较字符串列表和映射来过滤元素

Question

我有两个函数，它们获取字符串列表和映射作为参数并返回整数列表。如图：

def f1(names: List[String], namemap: Map[String, List[Int]]): List[Int] = 

def f2(names: List[String], namemap: Map[String, List[Int]]): List[Int] =

为了让下面的解释更容易理解，这里举个例子：

val nameslist = List("alex", "toby")

val namesmap = Map("sarah" -> List(5), "toby" -> List(6), "alex" -> List(1, 6))

println(f1(nameslist, namesmap))
//should print: List(6,1)
println(f2(nameslist, namesmap))
//should print: List(6)

函数f1 应该“比较”吗？列表namelist 中的字符串与映射namesmap 中的字符串，并返回找到字符串的namesmap 中的所有整数值。

函数f2 应该只返回在两个字符串中都找到的整数。

Answer 1

这应该会有所帮助。

val nameslist = List("alex", "toby", "jo")
val namesmap =
  Map("sarah" -> List(5), "toby" -> List(6), "alex" -> List(1, 6))

nameslist.flatMap(namesmap.get)  //List[List[Int]]
         .flatten                //List[Int]
         .distinct               //duplicates removed
//res0: List[Int] = List(1, 6)

nameslist.flatMap(namesmap.get)       //List[List[Int]]
         .reduceOption(_ intersect _) //Option[List[Int]]
         .getOrElse(Nil)              //List[Int]
//res1: List[Int] = List(6)