泛函编程(2)-初次体验泛函编程

来源:互联网 发布:讷河监狱事件 知乎 编辑:程序博客网 时间:2024/04/28 08:05

    泛函编程和数学方程式解题相似;用某种方式找出问题的答案。泛函编程通用的方式包括了模式匹配(pattern matching)以及递归思维(Recursive thinking)。我们先体验一下:(在阅读本系列博客文章之前,相信读者已经对Scala语言及REPL用法有所了解了。在这就不去解释Scala的语法语意了。)

先来个简单的:

def reportError(msgId: Int): String = msgId match {     | case 1 => "Error number 1."     | case 2 => "Error number 2."     | case 3 => "Error number 3."     | case _ => "Unknown error!"     | }reportError: (msgId: Int)String

很明显,这个函数的是一个纯函数,也是一个完整函数。因为函数主体涵盖了所有输入值(注意: case _ =>)。我们可以预知任何输入msgId值所产生的结果。还有,函数中没有使用任何中间变量。看看引用情况:

reportError(2)res3: String = Error number 2.scala> reportError(-1)res4: String = Unknown error!

恰如我们预测的结果。


再来看看一个递归(Recursion)例子:阶乘(Factorial)是一个经典样例:

def factorial(n: Int): Int = {  if ( n == 1) n  else n * factorial(n-1)  }                                               //> factorial: (n: Int)Int  factorial(4)                                    //> res48: Int = 24


也可以用模式匹配方式:

def factorial_1(n: Int): Int = n match {  case 1 => 1  case k => k * factorial(n-1)  }                                               //> factorial_1: (n: Int)Int  factorial_1(4)                                  //> res49: Int = 24

用模式匹配方式使函数意思表达更简洁、明了。


我们试着用“等量替换”方式逐步进行约化(reduce)

factorial(4)  4 * factorial(3)  4 * (3 * factorial(2))  4 * (3 * (2 * factorial(1)))  4 * (3 * (2 * 1)) = 24

可以得出预料的答案。

递归程序可以用 loop来实现。主要目的是防止堆栈溢出(stack overflow)。不过这并不妨碍我们用递归思维去解决问题。 阶乘用while loop来写:

def factorial_2(n: Int): Int = { var k: Int = n var acc: Int = 1 while (k > 1) { acc = acc * k; k = k -1} acc  }                                               //> factorial_2: (n: Int)Int  factorial_2(4)                                  //> res50: Int = 24

注意factorial_2使用了本地变量k,acc。虽然从表达形式上失去了泛函编程的优雅,但除了可以解决堆栈溢出问题外,运行效率也比递归方式优化。但这并不意味着完全违背了“不可改变性”(Immutability)。因为变量是锁定在函数内部的。

最后,也可用tail recursion方式编写阶乘。让编译器(compiler)把程序优化成改变成 loop 款式:

def factorial_3(n: Int): Int = {    @annotation.tailrec  def go(n: Int, acc: Int): Int = n match {  case 1 => acc  case k => go(n-1,acc * k)  }  go(n,1)  }                                               //> factorial_3: (n: Int)Int  factorial_3(4)                                  //> res51: Int = 24

得出的同样是正确的答案。这段程序中使用了@annotation.tailrec。如果被标准的函数不符合tail recusion的要求,compiler会提示。



0 0
原创粉丝点击