记忆用法/缓存存储

时间:2016-02-09 07:54:27

标签: python caching memoization

我正在编写一个程序来计算两个变量的Pascal身份,硬编码到程序中,因为我是Python新手并尝试缓存和memoization。以下是我到目前为止的情况:

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值?

谢谢。

1 个答案:

答案 0 :(得分:0)

让我们看看。首先,您具有两个变量的函数,但仅通过一个参数将结果存储在缓存中。因此,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
祝你好运! :)