第十章 Scala 容器基础(二十八)：集合排序

Problem

你想要对一个集合元素进行排序。或者你想定义一个自定义类来实现Ordered trait，来让你可以使用sorted方法，或者使用比较操符<,<=,>,>=来对类的实例进行比较。

Solution

你可以使用sorted或者sortWith方法来对集合进行排序。

Sorted方法可以对集合元素类型为Double，Float，Int和其他可以隐试转化scala.math.Ordering的进行排序。

scala> val l = List(10, 5, 8, 1, 7).sorted
l: List[Int] = List(1, 5, 7, 8, 10)

scala> val b = List("banana", "pear", "apple", "orange").sorted
b: List[String] = List(apple, banana, orange, pear)

Rich版本的numeric类(比如RichInt)和StringOps类都实现了Ordered trait，所以他们可以使用sorted方法实现排序。

SortWith方法让你可以使用自己的排序逻辑来实现排序规则。下面的例子展示了如何对集合元素类型为Int和String使用sortWith排序：

scala> List(10, 5, 8, 1, 7).sortWith(_ < _)
res14: List[Int] = List(1, 5, 7, 8, 10)

scala> List(10, 5, 8, 1, 7).sortWith(_ > _)
res15: List[Int] = List(10, 8, 7, 5, 1)

scala> List("banana", "pear", "apple", "orange").sortWith(_ < _)
res16: List[String] = List(apple, banana, orange, pear)

scala>  List("banana", "pear", "apple", "orange").sortWith(_ > _)
res17: List[String] = List(pear, orange, banana, apple)

你的排序方法的复杂度取决于你的排序需求。举个例子，你可以通过sort访问元素的方法，比如下面这个例子，按长度对一个字符串集合进行排序：

scala> List("banana", "pear", "apple", "orange").sortWith(_.length < _.length)
res18: List[String] = List(pear, apple, banana, orange)

scala> List("banana", "pear", "apple", "orange").sortWith(_.length > _.length)
res19: List[String] = List(banana, orange, apple, pear)

如果你的排序方法非常复杂或者会被重复使用，那么你可以先定义这个方法后，再调用此方法：

scala> def sortByLength(s1: String, s2: String) = {
     |   println("compare %s and %s".format(s1, s2))
     |   s1.length > s2.length
     | }
sortByLength: (s1: String, s2: String)Boolean

scala> List("banana", "pear", "apple").sortWith(sortByLength)
compare pear and banana
compare banana and pear
compare apple and pear
compare apple and pear
compare apple and banana
compare banana and apple
res20: List[String] = List(banana, apple, pear)

Discussion

如果你定义的类，没有定义对Ordering的隐式转换，那么你就没有办法通过调用sorted方法来对集合元素进行排序。

scala> class Person(var name: String) {
     |   override def toString = name
     | }
defined class Person

创建一个Person集合：

scala> val ty = new Person("Tyler")
ty: Person = Tyler

scala> val al = new Person("Al")
al: Person = Al

scala> val paul = new Person("Paul")
paul: Person = Paul

scala> val dudes = List(ty, al, paul)
dudes: List[Person] = List(Tyler, Al, Paul)

如果你调用sorted方法对dudes进行排序，那么你会看到下面的错误提示：

scala> dudes.sorted
<console>:13: error: No implicit Ordering defined for Person.
              dudes.sorted
                    ^

但是你可以使用sortWith对dudes进行排序：

scala> dudes.sortWith(_.name < _.name)
res1: List[Person] = List(Al, Paul, Tyler)

scala> dudes.sortWith(_.name > _.name)
res2: List[Person] = List(Tyler, Paul, Al)

Mix in the Ordered trait

混入Ordered特质能够让你的程序使用sorted方法来对Person集合进行排序，但是你必须实现compare方法。

class Person(var name: String) extends Ordered[Person]{
  override def toString = name

  override def compare(that: Person): Int = {
    if (this.name == that.name) return 0
    else if (this.name > that.name) return 1
    else return -1
  }
}

这个新的Person类就可以使用sorted方法来进行排序了。

Compare方法提供了排序功能，compare方法会这样进行工作：

如果两个对象相等，返回0
如果this<that那么返回一个负数
如果shit>that那么返回一个正数

类的两个实例谁大谁小完全取决于你的compare算法，因为目前这个算法仅仅比较两个字符串的值，所以也可以写成这样：

def compare (that: Person) = this.name.compare(that.name)

使用Ordered特质的另外一个好处是它可以让你在代码中直接比较对象实例。

if (al > ty) println("Al") else println("Tyler")

上面代码之所以可以工作是因为Ordered特质实现了<=, <, >, >=方法，并调用你定义的compare方法来使之生效。

第十章 Scala 容器基础(二十八)：集合排序

Problem

Solution

Discussion

Mix in the Ordered trait

See Also