我正在编写一个程序来计算两个变量的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
值?
谢谢。
答案 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
祝你好运! :)