HOF高阶函数

HOF高阶函数

函数也是值，就像其他类型的值，比如整形、字符串、列表；函数也可以赋值给一个变量、存储在一个数据结构里、像参数一样传递给另一个函数。

把一个函数当做参数传递给另一个函数在纯函数式编程里很有用，它被称为高阶函数（higher-order function，HOF）。

import scala.annotation.tailrec

object Hof {

  def factorial(n: Int): Int = {
    @tailrec
    def go(n: Int, acc: Int): Int = {
      if (n <= 0) acc
      else go(n - 1, n * acc)
    }
    go(n, 1)
  }

  def formatFactorial(n: Int): Unit ={
    val msg = "The factorial of %d is %d."
    println(msg.format(n, factorial(n)))
  }

  def formatResult(name: String, n: Int, f: Int => Int) ={
    val msg = "The %s of %d is %d."
    println(msg.format(name, n, f(n)))
  }

  def main(args: Array[String]): Unit = {
    formatFactorial(7)
    formatResult("factorial", 7, factorial)
  }
}

formatResult是一个HOF，它接受一个函数f为参数。我们给f函数声明一个类型，就像其他参数那样，它的类型是 Int=>Int（int to int），表示f接收一个整形参数并返回一个整形参数。