Python 字典 setdefault() 方法用作递归函数的返回值

Question

假设您想获得第 n 个斐波那契数。那么，一种可能性是使用递归函数

def Fib(n, d):
    """Assumes n is an int >= 0, d dictionary
    Returns nth Fibonacci number"""
    if n in d:
        return d[n]
    else:
        d[n] = Fib(n-1, d) + Fib(n-2, d)
        return d[n]

这很好用。我试图将其缩短为

 def Fib(n, d):
        return d.setdefault(n, Fib(n-1, d) + Fib(n-2, d))

但是当我可以调用它时

d={0:1, 1:1}
print(f(2, d))

，甚至使用 f(1,d)，它进入无限循环，并重新启动内核。事实上，任何这种形式的函数，比如说

def f(n, d):
    return d.setdefault(n, f(n-1,d))

有同样的问题。当我尝试调试它时，我看到 n 不断减小，通过值 1。我想我不明白这个方法的实现。我假设 setdefault 方法首先检查 key 是否在字典中并返回值，如果没有，则将默认值分配给 key 并返回默认值。我在这里缺少什么？（我使用 Python 3.9.1 和 Spyder 4.2.0）

Answer 1

你仍然需要一个基本情况，否则没有什么可以阻止它计算，fib(-1)，fib(-2)，fib(-99)，...

def fib(n, d):
  return n if n < 2 else d.setdefault(n, fib(n-1, d) + fib(n-2, d))

print(fib(10, {0:0, 1:1}))

55

setdefault 遇到的问题是 python 是一种应用命令语言。这意味着函数参数在调用函数之前进行评估。对于setdefault，我们将评估fib(n-1,d) + fib(n-2,d)在我们尝试在d 中查找n。

更好的接口可能是dict.setdefault(key, lambda: somevalue)，如果需要设置默认值，则仅执行 lambda。我们可以在下面写成lazydefault -

def lazydefault(d, key, lazyvalue):
  if key not in d:
    d[key] = lazyvalue()
  return d[key]

def fib(n, d):
  return lazydefault(d, n, lambda: fib(n-1, d) + fib(n-2, d))

print(fib(10, {0:0, 1:1}))

55