力扣算法打卡第24天(回溯) | 回溯算法理论基础、77. 组合

这篇博客介绍了如何运用回溯算法解决组合问题,如在N个数中找出k个数的所有可能组合。通过详细解释回溯算法的三部曲(定义终止条件、处理中间状态和递归)并展示具体的Python代码实现,博主展示了如何有效地实现这一过程。同时,还提到了剪枝技巧以提高算法效率,例如限制选择的范围以避免无效的搜索。
摘要由CSDN通过智能技术生成

回溯法,一般可以解决如下几种问题:

组合问题:N个数里面按一定规则找出k个数的集合
切割问题:一个字符串按一定规则有几种切割方式
子集问题:一个N个数的集合里有多少符合条件的子集
排列问题:N个数按一定规则全排列,有几种排列方式
棋盘问题:N皇后,解数独等等

回溯算法可以抽象为一棵树,宽度为每个结点处理的集合的大小(for循环),深度为递归的深度。
模板:

void backtracking(参数) {
    if (终止条件) {
        存放结果;
        return;
    }

    for (选择:本层集合中元素(树中节点孩子的数量就是集合的大小)) {
        处理节点;
        backtracking(路径,选择列表); // 递归
        回溯,撤销处理结果
    }
}

77. 组合

回溯算法三部曲

题目链接:力扣
思路:回溯

class Solution:
    
    def combine(self, n: int, k: int) -> List[List[int]]:
        path = []  # 当前递归路径上所搜集的元素
        res = []  # 结果集二维数组
        def backtracking(n, k, startIndex):  # ①确定函数参数,startindex确定本次搜索的起始位置
            if len(path) == k:  # ②确定终止条件
                res.append(path[:])  # 列表的深拷贝,注意不能直接res.append(path)
                print(path)
                return
            for i in range(startIndex, n+1):  # ③确定中间处理流程
                path.append(i)
                backtracking(n, k, i+1)
                path.pop()
        backtracking(n, k, 1)
        return res

剪枝:
i 至多等于 n - (k - len(path)) + 1,k-len(path)为还需要多少个元素才满足k,最后还要加1,比如n=5,k=3,此时len(path)为1,i至多从5-(3-1)+1=4开始,因为path还缺2个元素满足k,从5开始就不行了。

class Solution:
    
    def combine(self, n: int, k: int) -> List[List[int]]:
        path = []  # 当前递归路径上所搜集的元素
        res = []  # 结果集二维数组
        def backtracking(n, k, startIndex):  # 确定函数参数,startindex确定本次搜索的起始位置
            if len(path) == k:  # 确定终止条件
                res.append(path[:])  # 列表的深拷贝,注意不能直接res.append(path)
                print(path)
                return

            # 剪枝,i至多等于n-(k-len(path))+1,k-len(path)为还需要多少个元素才满足k
            for i in range(startIndex, n-(k-len(path))+1 +1):   
                path.append(i)
                backtracking(n, k, i+1)
                path.pop()
        backtracking(n, k, 1)
        return res
             


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值