最近,我编写了一个函数来生成具有非平凡约束的特定序列。这个问题有一个自然的递归解决方案。现在,即使对于相对较小的输入,序列也有数千个,因此我更喜欢使用我的算法作为生成器,而不是使用它来填充所有序列的列表。
下面是一个例子。假设我们想用一个递归函数计算一个字符串的所有排列。下面的朴素算法接受一个额外的参数'storage‘,并在找到时将一个排列附加到该参数后面:
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
此代码不起作用(该函数的行为类似于空生成器)。
我是不是遗漏了什么?有没有一种方法可以把上面的递归算法变成一个生成器,而不用迭代算法来代替它
https://stackoverflow.com/questions/248830
复制相似问题