如何将递归函数转换为生成器？

Question

我具有以下递归函数，用于为（命名的）职位列表生成有效配置的列表，其中每个职位只能使用一次：

def generate_configurations(configurations, named_positions, current):
    if len(current) == len(named_positions):
        configurations.append(current)
        return configurations

    name, positions = named_positions[len(current)]
    for x in positions:
        if x not in current:
            generate_configurations(configurations, named_positions, current + (x,))

    return configurations

这是我如何称呼它的一个例子：

named_positions = [('a', [0,1,2]),
                   ('b', [1,3]),
                   ('c', [1,2])]

for comb in generate_configurations([], named_positions, ()):
    print comb

给出以下输出：

(0, 1, 2)
(0, 3, 1)
(0, 3, 2)
(1, 3, 2)
(2, 3, 1)

同样，可能没有有效的组合，例如， named_positions = [('a', [3]), ('b', [3])] 。

现在，根据输入的named_positions ， configurations列表可能很快变得巨大，从而导致MemoryError 。 我相信可以将此函数重写为生成器，因此我尝试了以下操作：

def generate_configurations(named_positions, current):
    if len(current) == len(named_positions):
        yield current

    name, positions = named_positions[len(current)]
    for x in positions:
        if x not in current:
            generate_configurations(named_positions, current + (x,))

named_positions = [('a', [0,1,2]),
                   ('b', [1,3]),
                   ('c', [1,2])]

for comb in generate_configurations(named_positions, ()):
    print comb

但这根本不会产生任何结果。 我究竟做错了什么？

Answer 1

您需要yield递归调用堆栈，否则内部yield永远不会发生并被丢弃。 由于此标记为Python 2.7，因此可以通过更改以下内容来处理递归调用：

if x not in current:
    # Creates the generator, but doesn't run it out to get and yield values
    generate_configurations(named_positions, current + (x,))

至：

if x not in current:
    # Actually runs the generator and yields values up the call stack
    for y in generate_configurations(named_positions, current + (x,)):
        yield y

在Python 3.3及更高版本中，您可以直接通过以下方式委派：

if x not in current:
    yield from generate_configurations(named_positions, current + (x,)):

Answer 2

使用生成器时，需要确保子生成器递归调用传递回调用方法。

def recur_generator(n):
    yield my_thing
    yield my_other_thing
    if n > 0:
        yield from recur_generator(n-1)

注意这里的yield from是将yield调用传递回父调用的结果。

您应该将递归调用行更改为

yield from generate_configurations(named_positions, current + (x,))

否则，您的发电机就可以了。

编辑：没注意到这是python2。 您可以使用

for x in recur_generator(n-1):
    yield x

而不是的yield from 。