Scala 柯里化函數(shù)

柯里化函數(shù)

前面我們說過，Scala 允許程序員自己新創(chuàng)建一些控制結構，并且可以使得這些控制結構在語法看起來和 Scala 內(nèi)置的控制結構一樣，在 Scala 中需要借助于柯里化(Currying)，柯里化是把接受多個參數(shù)的函數(shù)變換成接受一個單一參數(shù)（最初函數(shù)的第一個參數(shù)）的函數(shù)，并且返回接受余下的參數(shù)而且返回結果的新函數(shù)的技術。

下面先給出一個普通的非柯里化的函數(shù)定義，實現(xiàn)一個加法函數(shù)：

scala> def plainOldSum(x:Int,y:Int) = x + y
plainOldSum: (x: Int, y: Int)Int

scala> plainOldSum(1,2)
res0: Int = 3

下面在使用“柯里化”技術來定義這個加法函數(shù)，原來函數(shù)使用一個參數(shù)列表，“柯里化”把函數(shù)定義為多個參數(shù)列表：

scala> def curriedSum(x:Int)(y:Int) = x + y
curriedSum: (x: Int)(y: Int)Int

scala> curriedSum (1)(2)
res0: Int = 3

當你調(diào)用 curriedSum (1)(2)時，實際上是依次調(diào)用兩個普通函數(shù)（非柯里化函數(shù)），第一次調(diào)用使用一個參數(shù) x，返回一個函數(shù)類型的值，第二次使用參數(shù)y調(diào)用這個函數(shù)類型的值，我們使用下面兩個分開的定義在模擬 curriedSum 柯里化函數(shù)：

首先定義第一個函數(shù)：

scala> def first(x:Int) = (y:Int) => x + y
first: (x: Int)Int => Int

然后我們使用參數(shù)1調(diào)用這個函數(shù)來生成第二個函數(shù)（回憶前面定義的閉包）。

scala> val second=first(1)
second: Int => Int = <function1>

scala> second(2)
res1: Int = 3

first，second的定義演示了柯里化函數(shù)的調(diào)用過程，它們本身和 curriedSum 沒有任何關系，但是我們可以使用 curriedSum 來定義 second，如下：

scala> val onePlus = curriedSum(1)_
onePlus: Int => Int = <function1>

下劃線“_” 作為第二參數(shù)列表的占位符， 這個定義的返回值為一個函數(shù)，當調(diào)用時會給調(diào)用的參數(shù)加一。

scala> onePlus(2)
res2: Int = 3

通過柯里化，你還可以定義多個類似 onePlus 的函數(shù)，比如 twoPlus

scala> val twoPlus = curriedSum(2) _
twoPlus: Int => Int = <function1>

scala> twoPlus(2)
res3: Int = 4