Day24-[回溯算法]开篇引入

阿琛与树

已于 2023-01-04 10:05:06 修改

阅读量149

点赞数

分类专栏：我学算法文章标签：算法 leetcode 职场和发展

于 2023-01-04 10:04:50 首次发布

本文链接：https://blog.csdn.net/championkai/article/details/128543364

版权

我学算法专栏收录该内容

23 篇文章 0 订阅

订阅专栏

文章介绍了如何使用回溯算法解决组合问题，展示了未优化和优化后的Python代码实现。优化主要体现在for循环的范围上，通过减少无效的遍历来提升效率。回溯算法的核心是横向遍历和纵向递归，同时进行了路径剪枝以提高性能。

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

代码随想录算法训练营Day24

77. Combinations

回溯算法核心精神: for循环横向遍历，递归纵向遍历，回溯不断调整结果集。

class Solution:
    def combine(self, n: int, k: int) -> List[List[int]]:
        # 提示使用回溯算法
        res = []
        path = []
        def backtrack(n, k, StartIndex):
            if len(path) == k:
                res.append(path[:])
                return
            for i in range(StartIndex, n + 1):
                path.append(i)
                backtrack(n, k, i+1)
                path.pop() # 回溯
        backtrack(n, k, 1)
        return res

剪枝优化:

class Solution:
    def combine(self, n: int, k: int) -> List[List[int]]:
        res=[]  #存放符合条件结果的集合
        path=[]  #用来存放符合条件结果
        def backtrack(n,k,startIndex):
            if len(path) == k:
                res.append(path[:])
                return
            for i in range(startIndex,n-(k-len(path))+2):  #优化的地方
                path.append(i)  #处理节点 
                backtrack(n,k,i+1)  #递归
                path.pop()  #回溯，撤销处理的节点
        backtrack(n,k,1)
        return res