如何解决如何在Python中使用递归编织两个列表
我想编织两个列表并输出所有可能的结果。
例如,
输入:两个列表l1 = [1,2],l2 = [3,4]
输出:[1、2、3、4],[1、3、2、4],[1、3、4、2],[3、1、2、4],[3、1、4、2 ],[3,4,1,2]
注意:我需要在每个列表中保持顺序(例如1始终在2之前,而3始终在4之前)
我解决这个问题的方法是从一个列表中删除头部,递归,然后对另一个列表执行相同的操作。代码如下:
all_possibles = []
def weaveLists(first,second,added):
if len(first) == 0 or len(second) == 0:
res = added[:]
res += first[:]
res += second[:]
all_possibles.append(res)
return
cur1 = first[0]
added.append(cur1)
first = first[1:]
weaveLists(first,added)
added = added[:-1]
first = [cur1] + first
cur2 = second[0]
added.append(cur2)
second = second[1:]
weaveLists(first,added)
added = added[:-1]
second = [cur2] + second
weaveLists([1,2],[3,4],[])
print(all_possibles)
我得到的结果是: [[1、2、3、4],[1、3、2、4],[1、3、4、2],[1、3、1、2、4],[1、3、1 4,2],[1、3、1、4、1、2]]
我不知道为什么对于最后三个列表,第一个列表的标题1未被删除。
有人可以帮忙吗?谢谢!
解决方法
这是另一种方法:我们在编织列表内生成第一个列表项的可能索引,并相应地填充列表。
我们可以使用itertools.combinations
生成索引:它是编织列表的索引的组合,每次取len(first_list)个。
from itertools import combinations
def weave(l1,l2):
total_length = len(l1) + len(l2)
# indices at which to put items from l1 in the weaved output
for indices in combinations(range(total_length),r=len(l1)):
out = []
it1 = iter(l1)
it2 = iter(l2)
for i in range(total_length):
if i in indices:
out.append(next(it1))
else:
out.append(next(it2))
yield out
样品运行:
l1 = [1,2]
l2 = [3,4]
for w in weave(l1,l2):
print(w)
[1,2,3,4]
[1,4,2]
[3,1,4]
[3,2]
另一个具有较长列表的示例:
l1 = [1,5]
for w in weave(l1,5]
[1,5,5]
[3,2]
,
获得这些意外结果的原因是您在此位置突变了added
:
added.append(cur1)
...这将(无意间)影响呼叫者的 added
列表。当“撤消”操作不是 更改列表时:
added = added[:-1]
这将创建一个新列表,因此此“撤消”操作不会不回滚呼叫者列表中的更改。
简单的解决方法是将对append
的呼叫替换为:
added = added + [cur1]
在第二个区块中应该发生同样的事情。
如果您即时传递递归调用的新值,然后用以下代码替换这两个代码块,则会更容易:
weaveLists(first[1:],second,added + [first[0]])
weaveLists(first,second[1:],added + [second[0]])
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。