从记忆子集和算法中获取组合？

Question

我一直在研究一个基本的子集和问题。给定一个和，比如说6，和数字[1,2,3,4,5,6]），我必须找到总的组合数s（对于s=6：[1,5]，[2,4]，[1,2,3]）。

我已经能够用蛮力解决这个问题，但还没有找到记忆它的方法，所以我的代码对于足够大的n值是不可行的。

我在这里找到了一个非常有效的记忆算法，但是它只给我组合的数量（所以，对于s=6，组合的数量将是3）——而不是组合本身。是否有可能既记忆这个问题（以便我可以对非常大的s值运行它）又能够输出组合本身？

Answer 1

您可以使用itertools简化问题。组合（）

>>> from itertools import combinations
>>> s = 6
>>> my_list = range(1, s)
# Value of 'my_list':
# [1, 2, 3, 4, 5]

>>> my_combinations = [combinations(my_list, i) for i in range(2, s)]
# Value of 'my_combinations' (in actual will be containg <combinations> objects):
# [[(1, 2), (1, 3), (1, 4), (1, 5), (2, 3), (2, 4), (2, 5), (3, 4), (3, 5), (4, 5)], [(1, 2, 3), (1, 2, 4), (1, 2, 5), (1, 3, 4), (1, 3, 5), (1, 4, 5), (2, 3, 4), (2, 3, 5), (2, 4, 5), (3, 4, 5)], [(1, 2, 3, 4), (1, 2, 3, 5), (1, 2, 4, 5), (1, 3, 4, 5), (2, 3, 4, 5)], [(1, 2, 3, 4, 5)]]

>>> my_required_set = [my_set for my_sublist in my_combinations for my_set in my_sublist if sum(my_set) == s]
>>> my_required_set
[(1, 5), (2, 4), (1, 2, 3)]