基本數(shù)據(jù)結(jié)構(gòu)

Scala提供了一些不錯的集合。

參考 Effective Scala 對怎樣使用 集合的觀點。

列表 List

scala> val numbers = List(1, 2, 3, 4)
numbers: List[Int] = List(1, 2, 3, 4)

集 Set

集沒有重復(fù)

scala> Set(1, 1, 2)
res0: scala.collection.immutable.Set[Int] = Set(1, 2)

元組 Tuple

元組是在不使用類的前提下，將元素組合起來形成簡單的邏輯集合。

scala> val hostPort = ("localhost", 80)
hostPort: (String, Int) = (localhost, 80)

與樣本類不同，元組不能通過名稱獲取字段，而是使用位置下標來讀取對象；而且這個下標基于1，而不是基于0。

scala> hostPort._1
res0: String = localhost

scala> hostPort._2
res1: Int = 80

元組可以很好得與模式匹配相結(jié)合。

hostPort match {
  case ("localhost", port) => ...
  case (host, port) => ...
}

在創(chuàng)建兩個元素的元組時，可以使用特殊語法：->

scala> 1 -> 2
res0: (Int, Int) = (1,2)

參考 Effective Scala 對 解構(gòu)綁定 （“拆解”一個元組）的觀點。

映射 Map

它可以持有基本數(shù)據(jù)類型。

Map(1 -> 2)
Map("foo" -> "bar")

這看起來像是特殊的語法，不過不要忘了上文討論的->可以用來創(chuàng)建二元組。

Map()方法也使用了從第一節(jié)課學(xué)到的變參列表：Map(1 -> "one", 2 -> "two")將變?yōu)? Map((1, "one"), (2, "two"))，其中第一個參數(shù)是映射的鍵，第二個參數(shù)是映射的值。

映射的值可以是映射甚或是函數(shù)。

Map(1 -> Map("foo" -> "bar"))

Map("timesTwo" -> { timesTwo(_) })

選項 Option

Option 是一個表示有可能包含值的容器。

Option基本的接口是這樣的：

trait Option[T] {
  def isDefined: Boolean
  def get: T
  def getOrElse(t: T): T
}

Option本身是泛型的，并且有兩個子類： Some[T] 或 None

我們看一個使用Option的例子：

Map.get 使用 Option 作為其返回值，表示這個方法也許不會返回你請求的值。

scala> val numbers = Map(1 -> "one", 2 -> "two")
numbers: scala.collection.immutable.Map[Int,String] = Map((1,one), (2,two))

scala> numbers.get(2)
res0: Option[java.lang.String] = Some(two)

scala> numbers.get(3)
res1: Option[java.lang.String] = None

現(xiàn)在我們的數(shù)據(jù)似乎陷在Option中了，我們怎樣獲取這個數(shù)據(jù)呢？

直覺上想到的可能是在isDefined方法上使用條件判斷來處理。

// We want to multiply the number by two, otherwise return 0.
val result = if (res1.isDefined) {
  res1.get * 2
} else {
  0
}

我們建議使用getOrElse或模式匹配處理這個結(jié)果。

getOrElse 讓你輕松地定義一個默認值。

val result = res1.getOrElse(0) * 2

模式匹配能自然地配合Option使用。

val result = res1 match {
  case Some(n) => n * 2
  case None => 0
}

參考 Effective Scala 對使用Options的意見。

函數(shù)組合子（Functional Combinators）

List(1, 2, 3) map squared對列表中的每一個元素都應(yīng)用了squared平方函數(shù)，并返回一個新的列表List(1, 4, 9)。我們稱這個操作map 組合子。 （如果想要更好的定義，你可能會喜歡Stackoverflow上對組合子的說明。）他們常被用在標準的數(shù)據(jù)結(jié)構(gòu)上。

map

map對列表中的每個元素應(yīng)用一個函數(shù)，返回應(yīng)用后的元素所組成的列表。

scala> numbers.map((i: Int) => i * 2)
res0: List[Int] = List(2, 4, 6, 8)

或傳入一個部分應(yīng)用函數(shù)

scala> def timesTwo(i: Int): Int = i * 2
timesTwo: (i: Int)Int

scala> numbers.map(timesTwo _)
res0: List[Int] = List(2, 4, 6, 8)

foreach

foreach很像map，但沒有返回值。foreach僅用于有副作用[side-effects]的函數(shù)。

scala> numbers.foreach((i: Int) => i * 2)

什么也沒有返回。

你可以嘗試存儲返回值，但它會是Unit類型（即void）

scala> val doubled = numbers.foreach((i: Int) => i * 2)
doubled: Unit = ()

filter

filter移除任何對傳入函數(shù)計算結(jié)果為false的元素。返回一個布爾值的函數(shù)通常被稱為謂詞函數(shù)[或判定函數(shù)]。

scala> numbers.filter((i: Int) => i % 2 == 0)
res0: List[Int] = List(2, 4)

scala> def isEven(i: Int): Boolean = i % 2 == 0
isEven: (i: Int)Boolean

scala> numbers.filter(isEven _)
res2: List[Int] = List(2, 4)

zip

zip將兩個列表的內(nèi)容聚合到一個對偶列表中。

scala> List(1, 2, 3).zip(List("a", "b", "c"))
res0: List[(Int, String)] = List((1,a), (2,b), (3,c))

partition

partition將使用給定的謂詞函數(shù)分割列表。

scala> val numbers = List(1, 2, 3, 4, 5, 6, 7, 8, 9, 10)
scala> numbers.partition(_ %2 == 0)
res0: (List[Int], List[Int]) = (List(2, 4, 6, 8, 10),List(1, 3, 5, 7, 9))

find

find返回集合中第一個匹配謂詞函數(shù)的元素。

scala> numbers.find((i: Int) => i > 5)
res0: Option[Int] = Some(6)

drop & dropWhile

drop 將刪除前i個元素

scala> numbers.drop(5)
res0: List[Int] = List(6, 7, 8, 9, 10)

dropWhile 將刪除元素直到找到第一個匹配謂詞函數(shù)的元素。例如，如果我們在numbers列表上使用dropWhile奇數(shù)的函數(shù), 1將被丟棄（但3不會被丟棄，因為他被2“保護”了）。

scala> numbers.dropWhile(_ % 2 != 0)
res0: List[Int] = List(2, 3, 4, 5, 6, 7, 8, 9, 10)

foldLeft

scala> numbers.foldLeft(0)((m: Int, n: Int) => m + n)
res0: Int = 55

0為初始值（記住numbers是List[Int]類型），m作為一個累加器。

直接觀察運行過程：

scala> numbers.foldLeft(0) { (m: Int, n: Int) => println("m: " + m + " n: " + n); m + n }
m: 0 n: 1
m: 1 n: 2
m: 3 n: 3
m: 6 n: 4
m: 10 n: 5
m: 15 n: 6
m: 21 n: 7
m: 28 n: 8
m: 36 n: 9
m: 45 n: 10
res0: Int = 55

foldRight

和foldLeft一樣，只是運行過程相反。

scala> numbers.foldRight(0) { (m: Int, n: Int) => println("m: " + m + " n: " + n); m + n }
m: 10 n: 0
m: 9 n: 10
m: 8 n: 19
m: 7 n: 27
m: 6 n: 34
m: 5 n: 40
m: 4 n: 45
m: 3 n: 49
m: 2 n: 52
m: 1 n: 54
res0: Int = 55

flatten

flatten將嵌套結(jié)構(gòu)扁平化為一個層次的集合。

scala> List(List(1, 2), List(3, 4)).flatten
res0: List[Int] = List(1, 2, 3, 4)

flatMap

flatMap是一種常用的組合子，結(jié)合映射[mapping]和扁平化[flattening]。 flatMap需要一個處理嵌套列表的函數(shù)，然后將結(jié)果串連起來。

scala> val nestedNumbers = List(List(1, 2), List(3, 4))
nestedNumbers: List[List[Int]] = List(List(1, 2), List(3, 4))

scala> nestedNumbers.flatMap(x => x.map(_ * 2))
res0: List[Int] = List(2, 4, 6, 8)

可以把它看做是“先映射后扁平化”的快捷操作：

scala> nestedNumbers.map((x: List[Int]) => x.map(_ * 2)).flatten
res1: List[Int] = List(2, 4, 6, 8)

這個例子先調(diào)用map，然后可以馬上調(diào)用flatten，這就是“組合子”的特征，也是這些函數(shù)的本質(zhì)。

參考 Effective Scala 對flatMap的意見。

擴展函數(shù)組合子

現(xiàn)在我們已經(jīng)學(xué)過集合上的一些函數(shù)。

我們將嘗試寫自己的函數(shù)組合子。

有趣的是，上面所展示的每一個函數(shù)組合子都可以用fold方法實現(xiàn)。讓我們看一些例子。

def ourMap(numbers: List[Int], fn: Int => Int): List[Int] = {
  numbers.foldRight(List[Int]()) { (x: Int, xs: List[Int]) =>
    fn(x) :: xs
  }
}

scala> ourMap(numbers, timesTwo(_))
res0: List[Int] = List(2, 4, 6, 8, 10, 12, 14, 16, 18, 20)

為什么是List[Int]()？Scala沒有聰明到理解你的目的是將結(jié)果積聚在一個空的Int類型的列表中。

Map?

所有展示的函數(shù)組合子都可以在Map上使用。Map可以被看作是一個二元組的列表，所以你寫的函數(shù)要處理一個鍵和值的二元組。

scala> val extensions = Map("steve" -> 100, "bob" -> 101, "joe" -> 201)
extensions: scala.collection.immutable.Map[String,Int] = Map((steve,100), (bob,101), (joe,201))

現(xiàn)在篩選出電話分機號碼低于200的條目。

scala> extensions.filter((namePhone: (String, Int)) => namePhone._2 < 200)
res0: scala.collection.immutable.Map[String,Int] = Map((steve,100), (bob,101))

因為參數(shù)是元組，所以你必須使用位置獲取器來讀取它們的鍵和值。呃！

幸運的是，我們其實可以使用模式匹配更優(yōu)雅地提取鍵和值。

scala> extensions.filter({case (name, extension) => extension < 200})
res0: scala.collection.immutable.Map[String,Int] = Map((steve,100), (bob,101))