2

我正在用 Python 解决经典的硬币问题。这是我的实现。

def memo(fn):
    def helper(*args): # here, * indicate the fn take arbitrary number of argumetns
        d = {}
        if args in d:
            return d[args] # args is a tuple, immutable, hashable
        else:
            res = fn(*args) # here * expand a tuple as arguments
            d[args] = res
            return res
    return helper

@memo
def change(options, n):
    if n < 0 or options ==():
        return 0
    elif n == 0:
        return 1
    else:
        return change(options, n- options[0]) + change(options[1:], n)

事实证明，memoized 版本比原始版本还要慢！为什么？我的实施出了什么问题？

这是没有记忆的：

In [172]: %timeit change((50, 25, 10, 5, 1), 100)
100 loops, best of 3: 7.12 ms per loop

这是记忆：

In [170]: %timeit change((50, 25, 10, 5, 1), 100)
10 loops, best of 3: 21.2 ms per loop
4

1 回答 1

7

在您当前的代码中：

def memo(fn):
    def helper(*args):
        d = {}

每次调用装饰函数时，您都会创建一个新的“缓存”字典。难怪它变慢了！最小的修复是：d 

def memo(fn):
    d = {}
    def helper(*args):

但总的来说它可能会更整洁。我用：

def memo(func):
    def wrapper(*args):
        if args not in wrapper.cache:
            wrapper.cache[args] = func(*args)
        return wrapper.cache[args]
    wrapper.cache = {}
    return wrapper

这使得更容易访问装饰函数cache以进行错误修复等。

于 2015-04-25T07:18:29.563 回答