(算法总结)寻找组合数,求出从整数1到n中和为m的所有组合

采用背包问题原理,仅考虑具有最大的数字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]

猜你喜欢

转载自blog.csdn.net/little_fire/article/details/81013721