记忆使用/缓存存储

Question

我正在编写一个程序，计算两个变量的 Pascal 标识，硬编码到程序中，因为我是 Python 新手并尝试缓存和记忆。这是我到目前为止所拥有的：

counter = 0

call_cacheK = {}
   def callTest(n, k):
   global counter

   if n in call_cacheK:
       return call_cacheK[n]  
   if k == 0:
       return 1
   elif k == n:
       return 1 
   elif (1 <= k) and (k <= (n-1)):
       counter += 1
       #call_cacheK[n] = result
       result = ((callTest(n-1, k) + callTest(n-1, k-1)))
   print(result)
   return result

callTest(20, 11)
#167,960

我的函数将输出最终的真实答案，但会输出很多答案。我似乎不知道如何正确存储要在缓存中使用的值。

如何正确使用call_cacheK来存储我已经使用过的result值？

谢谢。

Answer 1

让我们看看。首先，您有两个变量的函数，但仅通过一个参数将结果存储在缓存中。所以callTest(20, 11)、callTest(20, 10)、callTest(20, 9) 将在您的缓存中产生一个结果。让我们稍微重写一下你的函数：

call_cacheK = {}

def callTest(n, k):
    if (n, k) in call_cacheK:
        return call_cacheK[(n, k)]  
    if k == 0:
        return 1
    elif k == n:
        return 1 
    elif (1 <= k) and (k <= (n-1)):
        result = ((callTest(n-1, k) + callTest(n-1, k-1)))
        call_cacheK[(n, k)] = result
    print(result)
    return result

是的，没有计数器变量，因为我不知道你为什么需要它:)

另外，据我判断使用print(result)，您可能使用的是Python3.x。如果是这样，您可以使用standard cache implementing：

from functools import lru_cache

@lru_cache(maxsize=None)
def callTest2(n, k):  
    if k == 0:
        return 1
    elif k == n:
        return 1 
    elif (1 <= k) and (k <= (n-1)):
        result = ((callTest2(n-1, k) + callTest2(n-1, k-1)))
    print(result)
    return result

祝你好运！ :)