华为od技术面:排列组合

题目:给定两个正整数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)

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值