递归函数
递归函数就是在函数内部调用自身的函数,如阶乘 n! ,用 fact(n)表示
def fact(n)
if n == 1:
return 1
return n * fact(n - 1)
递归函数的定义简单,逻辑清晰,理论上所有递归函数都可以改写为循环的方式,但是循环的逻辑不如递归函数清晰。
使用递归函数时,需要防止栈溢出,因为在计算机中, 函数的调用是通过 栈 这种数据结构来实现的,每次调用一个函数,栈就会加一层栈帧,当函数返回时减一层,由于栈的大小有限,所以递归函数调用次数过多,就会出现栈溢出。
>>> fact(1000)
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
File "<stdin>", line 4, in fact
...
File "<stdin>", line 4, in fact
RuntimeError: maximum recursion depth exceeded in comparison
可以通过 尾递归优化 来解决这个问题,尾递归就是返回的值里面没有表达式,我们可以考虑用另一个函数 fact_iter(num,product) 。 product 用来储存 n-1 之前所有的乘积
def fact_iter(num,product):
if num == 1:
return product
return fact_iter(num-1,n*product)
def fact(n)
return fact_iter(n,1)
由于num-1 和 n*product 在调用函数前就已经被计算,所以不影响函数调用。
如果针对尾递归做了优化,栈就不会增长,也就不会溢出,但是对于大多数语言没有针对尾递归做优化,所以还是会栈溢出。
练习
汉诺塔的移动可以用递归函数非常简单地实现。
请编写move(n, a, b, c)函数,它接收参数n,表示3个柱子A、B、C中第1个柱子A的盘子数量,然后打印出把所有盘子从A借助B移动到C的方法,例如:
思路
需要的是借助B把所有盘子从A移到C,对于n个盘子,需要先把最底下那个移到C,首先需要把n-1个盘子移动到B,再把最大的盘子从A移动到C,再把n-1个盘子移动到A,之后开始迭代
# -*- coding: utf-8 -*-
def move(n, a, b, c):
#方法一
if n == 1:
print(a, '-->', c)
return
move(n - 1, a, c, b)
print(a, '-->', c)
move(n - 1, b, a, c)
return
#方法二
if n == 1:
print(a, '-->', c)
else:
move(n - 1, a, c, b)
print(a, '-->', c)
move(n - 1, b, a, c)
# 期待输出:
# A --> C
# A --> B
# C --> B
# A --> C
# B --> A
# B --> C
# A --> C
move(3, 'A', 'B', 'C')