最近我写了一个函数来生成具有非平凡约束的某些序列.这个问题伴随着一种自然的递归解决方案.现在碰巧,即使对于相对较小的输入,序列也是几千个,因此我宁愿使用我的算法作为生成器而不是使用它来填充具有所有序列的列表.
这是一个例子.假设我们想用递归函数计算字符串的所有排列.以下天真算法需要额外的参数'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
Run Code Online (Sandbox Code Playgroud)
(请不要关心效率低下,这只是一个例子.)
现在我想将我的函数转换为生成器,即生成排列而不是将其附加到存储列表:
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
Run Code Online (Sandbox Code Playgroud)
此代码不能正常工作(该函数的行为像一个空发生器).
我错过了什么吗?有没有办法将上述递归算法转换为生成器而不用迭代算法替换它?
我天真地尝试创建一个递归生成器.没工作.这就是我做的:
def recursive_generator(lis):
yield lis[0]
recursive_generator(lis[1:])
for k in recursive_generator([6,3,9,1]):
print(k)
Run Code Online (Sandbox Code Playgroud)
我得到的只是第一项6.
有没有办法使这样的代码工作?基本上yield在递归方案中将命令转移到上面的级别?