最近,我编写了一个函数来生成具有重要约束的某些序列。这个问题是通过自然递归解决方案来解决的。现在,即使对于相对较小的输入,序列也有数千个,因此我更愿意使用我的算法作为生成器,而不是使用它来填充所有序列的列表。
这是一个例子。假设我们想用递归函数计算字符串的所有排列。以下朴素算法采用额外的参数“存储”,并在找到一个参数时向其附加排列:
def getPermutations(string, storage, prefix=""):
if len(string) == 1:
storage.append(prefix + string) # <-----
else:
for i in range(len(string)):
getPermutations(string[:i]+string[i+1:], storage, prefix+string[i])
storage = []
getPermutations("abcd", storage)
for permutation in storage: print permutation
(请不要关心效率低下,这只是一个例子。)
现在我想将我的函数变成一个生成器,即产生一个排列而不是将其附加到存储列表中:
def getPermutations(string, prefix=""):
if len(string) == 1:
yield prefix + string # <-----
else:
for i in range(len(string)):
getPermutations(string[:i]+string[i+1:], prefix+string[i])
for permutation in getPermutations("abcd"):
print permutation
这段代码的作用是not工作(该函数的行为就像一个空的生成器)。
我错过了什么吗?
有没有办法把上面的递归算法变成生成器无需用迭代替换它?