一、回溯算法基础知识:
1.回溯算法常见可解决的问题:
- 组合问题:N个数里面按一定规则找出k个数的集合
- 切割问题:一个字符串按一定规则有几种切割方式
- 子集问题:一个N个数的集合里有多少符合条件的子集
- 排列问题:N个数按一定规则全排列,有几种排列方式
- 棋盘问题:N皇后,解数独等等
2.如何理解:(回溯算法像是递归里面嵌套着for循环)
回溯算法解决的都是在集合中递归查找子集,集合的大小就构成了树的宽度,递归的深度就构成了树的深度。
3.模板:
回溯三部曲:(个人觉得就是N叉树的遍历)
- 回溯函数返回值以及参数
- 终止条件
- 遍历过程
二、leetcode题目 :
1.letcode题目 77. 组合 - 力扣(LeetCode)
未优化版:
class Solution:
def combine(self, n: int, k: int) -> List[List[int]]:
path = []
result = []
def dfs(n,k,s_index,path,result):
if len(path) == k:
result.append(path[:])
return
for i in range(s_index,n+1):
path.append(i)
dfs(n,k,i+1,path,result)
path.pop()
dfs(n,k,1,path,result)
return result
优化版(通过剪枝):(就是剪掉了横向遍历过程中长度已经不满足搜索出path所需要的长度的节点)
class Solution:
def combine(self, n: int, k: int) -> List[List[int]]:
path = []
result = []
def dfs(n,k,s_index,path,result):
if len(path) == k:
result.append(path[:])
return
for i in range(s_index,n-(k-len(path))+2):
path.append(i)
dfs(n,k,i+1,path,result)
path.pop()
dfs(n,k,1,path,result)
return result