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

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

阅读量：6245 次

发布时间：2019-06-22

本文共 3486 字，大约阅读时间需要 11 分钟。

Problem

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

Solution

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

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

scala> val l = List(10, 5, 8, 1, 7).sortedl: List[Int] = List(1, 5, 7, 8, 10)scala> val b = List("banana", "pear", "apple", "orange").sortedb: 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)Booleanscala> List("banana", "pear", "apple").sortWith(sortByLength)compare pear and bananacompare banana and pearcompare apple and pearcompare apple and pearcompare apple and bananacompare banana and appleres20: 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 = Tylerscala> val al = new Person("Al")al: Person = Alscala> val paul = new Person("Paul")paul: Person = Paulscala> val dudes = List(ty, al, paul)dudes: List[Person] = List(Tyler, Al, Paul)

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

scala> dudes.sorted
    
     :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方法来使之生效。

Problem

Solution

Discussion

Mix in the Ordered trait

See Also