代码随想录算法训练营day24 | 77. 组合

最新推荐文章于 2024-09-08 15:12:17 发布

sunflowers11

最新推荐文章于 2024-09-08 15:12:17 发布

阅读量284

点赞数 2

分类专栏：代码随想录文章标签：算法数据结构

本文链接：https://blog.csdn.net/sunflowers11/article/details/136168484

版权

代码随想录专栏收录该内容

53 篇文章 0 订阅

订阅专栏

本文详细介绍了回溯算法，包括for循环实现的横向遍历和backtracking的纵向遍历，以及如何在组合问题中通过剪枝优化搜索过程。通过Solution类展示了两种版本的backtracking方法，分别针对未剪枝和剪枝的情况。

摘要由CSDN通过智能技术生成

今天开始回溯

回溯的模板。for循环是横向遍历，backtracking为纵向遍历。

回溯法解决的问题都可以抽象为树形结构，树的宽度为集合的大小，树的深度为递归的深度

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

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

77. 组合

未剪枝

class Solution:
    def __init__(self):
        self.result = []
        self.path = []

    def backtracking(self, n, k, startIndex):
        if len(self.path) == k:
            self.result.append(self.path[:])
            return

        for i in range(startIndex, n+1):
            self.path.append(i)
            self.backtracking(n, k, i+1)
            self.path.pop()

    def combine(self, n: int, k: int) -> List[List[int]]:
        self.backtracking(n, k, 1)
        return self.result

剪枝，主要在于如果剩下的数量+path中的数量不够K个了，那就停止循环遍历

class Solution:
    def __init__(self):
        self.result = []
        self.path = []

    def backtracking(self, n, k, startIndex):
        if len(self.path) == k:
            self.result.append(self.path[:])
            return

        for i in range(startIndex, n-(k-len(self.path))+2):
            self.path.append(i)
            self.backtracking(n, k, i+1)
            self.path.pop()

    def combine(self, n: int, k: int) -> List[List[int]]:
        self.backtracking(n, k, 1)
        return self.result