所有人都听过这样一个歌谣:从前有座山,山里有座庙,庙里有个和尚在讲故事:从前有座山。。。。,虽然这个歌谣并没有一个递归边界条件跳出循环,但无疑地,这是递归算法最朴素的落地实现,本次我们使用Golang1.18回溯递归与迭代算法的落地场景应用。

递归思想与实现

递归思想并非是鲜为人知的高级概念,只不过是一种相对普遍的逆向思维方式,这一点我们在:人理解迭代,神则体会递归,从电影艺术到Python代码实现神的逆向思维模式中已经探讨过,说白了就是一个函数直接或者间接的调用自己,就是递归,本文开篇和尚讲故事的例子中,和尚不停地把他自己和他所在的庙和山调用在自己的故事中,因此形成了一个往复循环的递归故事,但这个故事有个致命问题,那就是停不下来,只能不停地讲下去,所以一个正常的递归必须得有一个递归边界条件,用来跳出无限递归的循环:

package main    import (  "fmt"  )    func story(n int) int {  if n <= 0 {  return 0  }  return story(n - 1)    }    func main() {    res := story(5)    fmt.Println(res)    }  

这里我们声明了一个故事函数,参数为n,即讲n遍同样的故事,并且调用自己,每讲一次n减1,即减少一次讲故事总数,但如果我们不设置一个递归边界条件,那么函数就会无限递归下去,所以如果n小于等于0了,那么我们就结束这个故事:

➜  mydemo git:(master) ✗ go run "/Users/liuyue/wodfan/work/mydemo/tests.go"  0

所以 if n <= 0 就是递归边界条件。

那么递归的底层是如何实现的呢?假设我们要针对n次故事做一个高斯求和:

package main    import (  "fmt"  )    func story(n int) int {  if n <= 0 {  return 0  }  return n + story(n-1)    }    func main() {    res := story(5)    fmt.Println(res)    }

程序输出:

➜  mydemo git:(master) ✗ go run "/Users/liuyue/wodfan/work/mydemo/tests.go"  15

那么这一次递归高斯求和函数的底层实现应该是这样:

5+story(4)  5+(4+ story(3))  5+(4+(3+ story(2)))  5+(4+(3+(2+ story(1))))  5+(4+(3+(2+1)))  15

当story函数每次被调用时,都会在内存中创建一个帧,来包含函数的局部变量和参数,对于递归函数,栈上可能同时存在多个函数帧。当每调用一次函数story(n)时,栈顶指针就会往栈顶移动一个位置,直到满足退出递归的条件(n<=0)之后再依次返回当前的结果直接,栈顶指针被压入栈底方向。

也就是说,内存栈会存储每一次递归的局部变量和参数,这也就是递归算法的性能被人们所诟病的原因,即不是自己调用自己而性能差,而是自己调用自己时,系统需要保存每次调用的值而性能差。

尾递归优化

尾递归相对传统的普通递归,其是一种特例。在尾递归中,先执行某部分的计算,然后开始调用递归,所以你可以得到当前的计算结果,而这个结果也将作为参数传入下一次递归。这也就是说函数调用出现在调用者函数的尾部,因为是尾部,所以其有一个优越于传统递归之处在于无需去保存任何局部变量,从内存消耗上,实现节约特性:

package main    import (  "fmt"  )func tail_story(n int, save int) int {    if n <= 0 {  return save  }  return tail_story(n-1, save+n)    }    func main() {    save := 0    res := tail_story(5, save)    fmt.Println(res)    }

程序返回:

➜  mydemo git:(master) ✗ go run "/Users/liuyue/wodfan/work/mydemo/tests.go"  15

可以看到,求和结果和普通递归是一样的,但过程可不一样:

tail_story(5,0)  tail_story(4,5)  tail_story(3,9)  tail_story(2,12)  tail_story(1,14)  tail_story(0,15)

因为尾递归通过参数将计算结果进行传递,递归过程中系统并不保存所有的计算结果,而是利用参数覆盖旧的结果,如此,就不会到处栈溢出等性能问题了。

递归应用场景

在实际工作中,我们当然不会使用递归讲故事或者只是为了计算高斯求和,大部分时间,递归算法会出现在迭代未知高度的层级结构中,即所谓的“无限极”分类问题:

package main    import (  "fmt"  )    type cate struct {  id   int  name string  pid  int  }    func main() {  allCate := []cate{  cate{1, "计算机课程", 0},  cate{2, "美术课程", 0},  cate{3, "舞蹈课程", 0},  cate{4, "Golang", 1},  cate{5, "国画", 2},  cate{6, "芭蕾舞", 3},  cate{7, "Iris课程", 4},  cate{8, "工笔", 5},  cate{9, "形体", 6},  }    fmt.Println(allCate)    }

程序输出:

[{1 计算机课程 0} {2 美术课程 0} {3 舞蹈课程 0} {4 Golang 1} {5 国画 2} {6 芭蕾舞 3} {7 Iris课程 4} {8 工笔 5} {9 形体 6}]

可以看到,结构体cate中使用pid来记录父分类,但展示的时候是平级结构,并非层级结构。

这里使用递归算法进行层级结构转换:

type Tree struct {  id   int  name string  pid  int  son  []Tree  }

新增加一个Tree的结构体,新增一个子集的嵌套属性。

随后建立递归层级结构函数:

func CategoryTree(allCate []cate, pid int) []Tree {  var arr []Tree  for _, v := range allCate {  if pid == v.pid {  ctree := Tree{}  ctree.id = v.id  ctree.pid = v.pid  ctree.name = v.name    sonCate := CategoryTree(allCate, v.id)    ctree.son = sonCate    arr = append(arr, ctree)  }  }  return arr  }

随后调用输出:

package main    import (  "fmt"  )    type cate struct {  id   int  name string  pid  int  }    type Tree struct {  id   int  name string  pid  int  son  []Tree  }    func CategoryTree(allCate []cate, pid int) []Tree {  var arr []Tree  for _, v := range allCate {  if pid == v.pid {  ctree := Tree{}  ctree.id = v.id  ctree.pid = v.pid  ctree.name = v.name    sonCate := CategoryTree(allCate, v.id)    ctree.son = sonCate    arr = append(arr, ctree)  }  }  return arr  }    func main() {  allCate := []cate{  cate{1, "计算机课程", 0},  cate{2, "美术课程", 0},  cate{3, "舞蹈课程", 0},  cate{4, "Golang", 1},  cate{5, "国画", 2},  cate{6, "芭蕾舞", 3},  cate{7, "Iris课程", 4},  cate{8, "工笔", 5},  cate{9, "形体", 6},  }    arr := CategoryTree(allCate, 0)  fmt.Println(arr)    }

程序返回:

[{1 计算机课程 0 [{4 Golang 1 [{7 Iris课程 4 []}]}]} {2 美术课程 0 [{5 国画 2 [{8 工笔 5 []}]}]} {3 舞蹈课程 0 [{6 芭蕾舞 3 [{9 形体 6 []}]}]}]

这里和Python版本的无限极分类:使用Python3.7+Django2.0.4配合vue.js2.0的组件递归来实现无限级分类(递归层级结构)有异曲同工之处,但很显然,使用结构体的Golang代码可读性更高。

结语

递归并非是刻板印象中的性能差又难懂的算法,正相反,它反而可以让代码更加简洁易懂,在程序中使用递归,可以更通俗、更直观的描述逻辑。