停止递归发生器&排列

时间:2010-08-10 11:03:24

标签: python exception recursion generator permutation

作为练习,我一直在尝试各种方法来生成Python中列表的所有排列 - 递归,非递归...... - 并将性能与itertools.permutations()进行比较。但我在使用递归方法的生成器版本时遇到问题,该方法没有StopIteration异常干净地完成,而是抛出IndexError

def spawnperms(alist):
    """same algorithm as recursive option, but a generator"""
    if (alist == []):
        yield []
    for perm in spawnperms(alist[:-1]):
        for i in range(len(perm)+1):
            yield perm[:i] + [alist[-1]] + perm[i:]

从Python解释器调用它:

>>> for i in spawnperms(range(3)):
...     print i
... 
[2, 1, 0]
[1, 2, 0]
[1, 0, 2]
[2, 0, 1]
[0, 2, 1]
[0, 1, 2]
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "<stdin>", line 5, in spawnperms
  File "<stdin>", line 5, in spawnperms
  File "<stdin>", line 5, in spawnperms
  File "<stdin>", line 7, in spawnperms
IndexError: list index out of range

哎哟。我尝试用pdb逐步完成它,这几乎在我的大脑中创建了一个堆栈溢出,但我所理解的是递归“向下”到空列表,然后是外部(我认为){{1循环用完了索引。

如何更正我的代码?

编辑:从Mark Byers那里学到的一个看似简单的正确答案是 干净的编码实践可以防止错误 。如果我在for之后系统地使用else,无论我是否认为可以重新审视这种情况,这都不会发生。它仍然感觉非常愚蠢!

1 个答案:

答案 0 :(得分:6)

您错过了else

if (alist == []):
    yield []
else:
    for ...

这是因为yield的行为方式与return不同。当您请求下一个值时,在yield语句后继续执行。