gpt4 book ai didi

python - 在 Python 上使用回溯的子集总和

转载 作者:太空宇宙 更新时间:2023-11-04 10:25:35 25 4
gpt4 key购买 nike

所以我想打印出初始集合的所有子集,这些子集加起来为 21。到目前为止,我只想到了这个

def twentyone(array, num=21):
if len(array) == 0:
return None
else:
if array[0] == num:
return [array[0]]
else:
with_v = twentyone(array[1:], (num - array[0]))
if with_v:
return [array[0]] + with_v
else:
return twentyone(array[1:], num)

它确实给出了解决方案,但只是第一个。我如何更改它以便它给我所有可能的子集。我试过做一些改变,但它只给了我嵌套列表。你能帮忙的话,我会很高兴。

最佳答案

你可以创建一个递归生成器:

def twentyone(array, num=21):
if num < 0:
return
if len(array) == 0:
if num == 0:
yield []
return
for solution in twentyone(array[1:], num):
yield solution
for solution in twentyone(array[1:], num - array[0]):
yield [array[0]] + solution

例子:

>>> list(twentyone([5, 16, 3, 2]))
[[16, 3, 2], [5, 16]]

关于python - 在 Python 上使用回溯的子集总和,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/29456502/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com