Scala 接受匿名比较器吗？

Question

我如何接受一个函数作为参数传递给 scala 中的 Vector.sortBy() ？

目前我有一个这样的函数：

private def buildTree(data: Vector[Data2D]): Node = {
   var sorted: Vector[Data2D] = data.sortBy(_.x)
   // etc...
 }

但是，我想传递“_.x”（对x排序）或“_.y”作为函数的参数，所以我可以这样做：

private def buildTree(data: Vector[Data2D], comparator): Node = {
   var sorted: Vector[Data2D] = data.sortBy(comparator)
   // etc...

   if(comparator == _.x){
       buildTree(data, _.y)
   }

 }

所以我会就像检查当前的“比较器”是什么，然后递归地将 y 坐标的比较器传递给它。

我希望这一点很清楚。在Java中我会把它写成：

private Node buildTree(List<Data2D> data, Comparator<Data2D> comparator) {
 // Sorted on x or y
 Collections.sort(data, comparator);

 // ... snip ...

 if (comparator instanceof XComparator) {
    // Recurse
    Node subtree = buildTree(data, YComparator.INSTANCE);
    ret.setSubtree(subtree);
 }
 return ret;
}

// Then build tree is called like:
Node root = tree.buildTree(data, XComparator.INSTANCE)

Answer 1

实际上，您想要一个Ordering。像这样：

class XOrdering extends Ordering[Data2D] {
    override def compare(x: Data2D, y: Data2D): Int = Ordering.Int(x.x, y.x)
}

class YOrdering extends Ordering[Data2D] {
    override def comapre(x: Data2D, y: Data2D): Int = Ordering.Int(x.y, y.y)
}

private def buildTree(data: Vector[Data2D], ordering: Ordering[Data2D]): Node = {
   var sorted: Vector[Data2D] = data.sorted(ordering)
   // etc...

   ordering match {
     case _: XOrdering => buildTree(data, YOrdering)
     case _: YOrdering => buildTree(data, XOrdering)
     case _ => error("I don't know what that ordering is!")
   }
 }

sortBy 方法只是为您创建一个Ordering。 sortBy(_.x) 相当于 sorted(Ordering.by(_.x))。

Answer 2

如果您检查文档，您会发现 sortBy 采用一个参数： f: A =>; B。看起来像一个通用函数，它接受 A 作为参数并生成 B。那么，让我们尝试一下：

val orderX = (d: Data2D) => d.x
val orderY = (d: Data2D) => d.y

它定义了我们想要的两个函数。现在我们可以调用

data.sortBy(orderer)

和测试

if (orderer==orderX) buildTree(data, orderY)

（尽管我建议将两个比较器作为参数传递，而不是搜索特定的常量）。