77. 组合
- 第一次专门做回溯,有点难理解。
- 首先可以理解回溯可以可视化为树的搜索,因此这道题,树的宽度为n,树的深度为k
- path作为一个参数传入有点难想
- 回溯没有返回值
- 剪纸更难想,通过列算式可以勉强得到for的表达式,但是要自己写出来还是有点难
class Solution:
def combine(self, n: int, k: int) -> List[List[int]]:
# 递归的话,就是取一个数,然后递归为k-1
self.output = []
self.backtracking(n, k, 1, [])
return self.output
# start_index表示从哪里开始遍历
def backtracking(self, n, k, start_index, path): # path用来记录遍历的路径
if len(path) == k: # 当路径长度==k时,输出,相当于遍历到了叶子节点
self.output.append(path[:]) # 为啥这里时path就不行?
for i in range(start_index, n+1):
path.append(i)
self.backtracking(n,k,i+1,path)
path.pop() # 回溯
剪枝后:
class Solution:
def combine(self, n: int, k: int) -> List[List[int]]:
result = [] # 存放结果集
self.backtracking(n, k, 1, [], result)
return result
def backtracking(self, n, k, startIndex, path, result):
if len(path) == k:
result.append(path[:])
return
for i in range(startIndex, n - (k - len(path)) + 2): # 优化的地方
path.append(i) # 处理节点
self.backtracking(n, k, i + 1, path, result)
path.pop() # 回溯,撤销处理的节点
需要多做做题,找找感觉