实现Python子集总和算法——找出集合中的元素相加等于给定值的所有子集

实现Python子集总和算法——找出集合中的元素相加等于给定值的所有子集

子集总和问题是一个经典的计算机科学问题，它涉及寻找集合中所有可能的子集，所有子集中元素的和等于给定值。本篇文章将介绍如何实现 Python 子集总和算法。我们将提供完整的 Python 代码以及一些示例输出，以便您更好地了解该算法。

算法实现

我们将使用递归方法来实现此算法。在每个递归级别中，我们都需要选择一个元素，同时考虑包含该元素和不包含该元素的情况。如果当前元素加上已经选择的元素的和小于目标和，则继续递归查找下一个元素。如果当前元素加上已经选择的元素的和等于目标和，则记录这个子集。最后，返回所有符合条件的子集。下面是完整的 Python 代码：

def subset_sum(array, target, partial=[], result=[]):
    s = sum(partial)

    if s == target:
        result.append(partial)

    if s >= target:
        return

    for i in range(len(array)):
        n = array[i]
        remaining = array[i+1:]
        subset_sum(remaining, target, partial + [n], result)

    return result