题目:给定两个正整数n,m,在[1,2,3,,,n]中选取m个数,输出m个数的所有组合
解题:使用回溯算法来生成组合列表,首先定义了一个结果列表res和一个从1到n的数字列表nums。
接下来定义了一个内部函数backtrack用于递归生成组合列表。
backtrack函数接受两个参数,一个是当前生成的组合列表temp,另一个是起始位置start。
当temp的长度等于m时,说明已经生成了一个长度为m的组合,将其添加到结果列表res中。
否则,从start开始遍历nums中的元素,将当前元素添加到temp中,然后递归调用backtrack函数,传入更新后的temp和i+1作为起始位置。
递归调用完成后,需要将temp中的最后一个元素弹出,以便下一次遍历可以选择其他元素。
最后返回生成的结果列表res。
这个函数可以用来解决组合生成的问题,例如从1到n的数字中选择m个数字的所有组合。
def combine(n, m):
res = []
def backtrack(temp, start):
if len(temp) == m:
res.append(temp[:])
else:
for i in range(start, n + 1):
temp.append(i)
backtrack(temp, i + 1)
temp.pop()
backtrack([], 1)
return res
n = 5
m = 3
result = combine(n, m)
for sublist in result:
print(sublist)