10

我在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 函数中,但我想找到一个更“功能风格”的解决方案,可能会返回一个像这样的元组:(找到元素,满足条件)。

在此先感谢您的帮助

4

3 回答 3

9

我会这样做:

斯卡拉 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)

斯卡拉 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)

view/确保映射将projection按需完成,而不是应用于整个列表。

于 2010-02-01T10:41:17.973 回答
3
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])或定义您自己的运算符。

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

于 2010-02-01T10:29:58.200 回答
1

这是丹尼尔(和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 的“投影”。)

于 2010-02-01T16:38:20.693 回答