采用背包问题原理,仅考虑具有最大的数字n是否存在与结果集合中,考虑以下两种情形:
(1)n在集合中,剩下的n-1个数字需要组成一个和为n-m的组合;
(2)n不在集合中,剩下的n-1个数字仍需要组成和为m的组合;
由于需要给出所有的组合可能,因此是一个回溯的过程。
算法设计思路:
由于是个回溯递归的过程,因此需要首先给出递归终止条件:当需要求和的数字小于等于0或所有数字都用完了的时候,就是程序终止的时候。
用一个列表数组item存储当前的候选集合,result列表存放最终的满足条件的结果,当满足条件时,将item加入result中。
因此,回溯的代码如下:
def find_combine(item, n, m, result):
if n <= 0 or m <= 0:
return
if n == m:
result.append(item + [n])
item.append(n)
find_combine(item, n-1, m-n, result)
item.pop()
find_combine(item, n-1, m, result)
用一组数字,运行一下,这里假设n=9,m=10
if __name__ == '__main__':
item = []
result = []
find_combine(item, 9, 10, result)
for r in result:
print(r)
运行结果如下:
[9, 1]
[8, 2]
[7, 3]
[7, 2, 1]
[6, 4]
[6, 3, 1]
[5, 4, 1]
[5, 3, 2]
[4, 3, 2, 1]