需求:
某些时候我们想为多个函数,统一添加某种功能,比如计时统计、日志记录、缓存运算结果等等
我们不想在每个函数内一一添加完全相同的代码,有什么好的解决方案?
思路:
定义装饰器函数,用它来生成一个在原函数基础上添加了新功能的函数,替代原函数
代码:
def memo(func):
cache = {}
def wrap(*args):
res = cache.get(args) # 此处args传入的是一个元组,元组可以作为字典的键
if not res:
res = cache[args] = func(*args) # 将传入的元组参数打散来
return res
return wrap
# [题目1] 斐波那契数列(Fibonacci sequence):
# F(0)=1,F(1)=1,F(n)=F(n-1)+F(n-2)(n>=2)
# 1,1,2,3,5,8,13,21,34,...
# 求数列第n项的值
# 使用缓存,可以去除重复的大量运算,提高运算的速度
@memo # 等价于fibonacci = memo(fibonacci)
def fibonacci(n):
if n <= 1:
return 1
res = fibonacci(n-1) + fibonacci(n-2)
return res
# [题目2] 走楼梯问题
# 有100阶楼梯,一个人可以迈1~3阶,一共有多少走法?
@memo
def climb(n,steps):
count = 0
if n == 0:
count = 1
elif n > 0:
for step in steps:
count += climb(n-step,steps)
return count
# fibonacci = memo(fibonacci) # 偷梁换柱,用户直接调用的还是fibonacci,不感知函数内部
print(fibonacci(50))
# climb = memo(climb) # 偷梁换柱,用户直接调用的还是climb,不感知函数内部
print(climb(100,(1,2,3)))