本文为大家分享了python利用高阶函数实现剪枝函数的具体代码,供大家参考,具体内容如下

案例:

某些时候,我们想要为多个函数,添加某种功能,比如计时统计,记录日志,缓存运算结果等等

需求:

在每个函数中不需要添加完全相同的代码

如何解决?

把相同的代码抽调出来,定义成装饰器

求斐波那契数列(黄金分割数列),从数列的第3项开始,每一项都等于前两项之和

求一个共有10个台阶的楼梯,从下走到上面,一次只能迈出1~3个台阶,并且不能后退,有多少中方法?

上台阶问题逻辑整理:

每次迈出都是 1~3 个台阶,剩下就是 7~9 个台阶

如果迈出1个台阶,需要求出后面9个台阶的走法

如果迈出2个台阶,需要求出后面8个台阶的走法

如果迈出3个台阶,需要求出后面7个台阶的走法

此3种方式走法,通过递归方式实现,递归像树,每次递归都生成子节点函数

以上两个问题通过递归来解决,就会出现一个问题,出现重复求解问题,把重复求解的过程剔除掉,在c++语言中称为剪枝函数

#!/usr/bin/python3
def jian_zhi(func):
# 中间字典,判断已经是否求解过
median = {}
def wrap(*args):
# 假如不在中间字典中,说明没有求解过,添加到字典中去,在的话,直接返回
if args not in median:
median[args] = func(*args)
return median[args]
return wrap
@jian_zhi
def fibonacci(n):
if n <= 1:
return 1
return fibonacci(n-1) + fibonacci(n-2)
@jian_zhi
def climb(n, steps):
count = 0
# 当最后台阶为0的时候,说明最后只是走了一次
if n == 0:
count = 1
# 当最后台阶不为0的时候,说明还需要走至少一次
elif n > 0:
# 对三种情况进行分别处理momo
for step in steps:
count += climb(n-step, steps)
# 返回每次递归的计数
return count
if __name__ == '__main__':
print(climb(10, (1, 2, 3)))
print(fibonacci(20))

所谓的剪枝函数不过是保证每次递归的函数唯一性,利用中间字典保存已经执行过得函数和参数,通过判断参数,剔除重复的函数调用