在 scala 列表中查找元素并知道满足哪个谓词

Question

我在 scala 中遇到以下问题。我必须找到 al 列表中的第一个元素，该元素满足具有 OR 中两个条件的谓词函数。问题是我想获取该元素，但也想知道这两个条件中的哪一个已得到满足。这是一个简单的例子：

val l1 = List("A", "B", "AA", "BB")
val l2 = List("AA", "BB", "A", "B")

def c1(s: String) = s.startsWith("B")
def c2(s: String) = s.length == 2

println(l1.find(s => c1(s) || c2(s)))
println(l2.find(s => c1(s) || c2(s)))

结果是：

Some(B)
Some(AA)

对于 l1 情况，我希望有一些返回值（例如字符串）表明 c1 得到满足（对于 l2 情况为 c2）。 一个可能的解决方案可能是在测试之前定义一个 var 并将其设置在 c1 和 c2 函数中，但我想找到一个更“函数式”的解决方案，也许返回一个元组，例如：（找到元素，满足条件）。

预先感谢您的帮助

Answer 1

我会这样做：

Scala 2.8：

def find2p[T](l: List[T], p1: T => Boolean, p2: T => Boolean) = 
  l.view.map(el => (el, p1(el), p2(el))).find(t => t._2 || t._3)

Scala 2.7：

def find2p[T](l: List[T], p1: T => Boolean, p2: T => Boolean) = 
  l.projection.map(el => (el, p1(el), p2(el))).find(t => t._2 || t._3)

视图/投影确保映射将按需完成，而不是应用于整个列表。

Answer 2

def find[T](l1 : List[T], c1 : T => Boolean, c2 : T => Boolean) = ((None : Option[(String, T)]) /: l1)( (l, n) => l match {
    case x : Some[_] => l
    case x if c1(n) => Some("c1", n)
    case x if c2(n) => Some("c2", n)
    case _ => None
})

scala> find(l1, c1, c2)
res2: Option[(String, java.lang.String)] = Some((c1,B))

scala> find(l2, c1, c2)
res3: Option[(String, java.lang.String)] = Some((c2,AA))

根据您的要求，您可以使用参数 Map[T ==> Boolean, String] 用于要返回的标签字符串： def find[T](l1 : List[T], fs : Map[T => Boolean, String]) 或定义您自己的运算符。

这将评估整个列表，其中查找中止找到的第一个元素。

Answer 3

这是 Daniel（和 Retronym）答案的变体。

如果您只想要成功的谓词（列表之外），那么您可以使用

def findP[T](list: Iterable[T], preds: Iterable[T=>Boolean]) = {
  list.view.map( x => (x , preds.find( _(x) )) ).find( _._2.isDefined )
}

或者，您可以使用命名谓词列表：

def findP[T](list: Iterable[T],preds: Iterable[(T=>Boolean,String)]) = {
  list.view.map(x => (x , preds.find( _._1(x) ))).find( _._2.isDefined )
}

scala> findP(
     |   List(1,2,3,4,5,6),
     |   List( ((i:Int)=>i>4,"Fred") , ((i:Int)=>(i%6)==0,"Barney"))
     | )
res2: Option[(Int, Option[((Int) => Boolean, String)])] =
  Some((5,Some((<function1>,Fred))))

结果有点混乱，但可以很容易地展开以准确给出您所要求的内容：

def findP[T](list: Iterable[T],preds: Iterable[(T=>Boolean,String)]) = {
  list.view.map(x => (x , preds.find( _._1(x) ))).find( _._2.isDefined ) match {
    case Some((i,Some((_,s)))) => Some((i,s))
    case _ => None
  }
}

（这是 2.8 的代码；将 2.7 的“视图”切换为“投影”。）