2023.6.26
端午节假后回归┭┮﹏┭┮
回溯,有点难,喵的
不剪枝
class Solution:
def combine(self, n: int, k: int) -> List[List[int]]:
self.result = []
self.backtracking(1, n, [], k)
return self.result
def backtracking(self, startindex, n, path, k):
if len(path) == k:
self.result.append(path[:]) # 收集结果
return
for i in range(startindex, n + 1):
path.append(i)
self.backtracking(i+1, n, path, k)
path.pop()
剪枝
class Solution:
def combine(self, n: int, k: int) -> List[List[int]]:
self.result = []
self.backtracking(1, n, [], k)
return self.result
def backtracking(self, startindex, n, path, k):
if len(path) == k:
self.result.append(path[:]) # 收集结果
return
for i in range(startindex, n - (k -len(path)) + 2):
path.append(i)
self.backtracking(i+1, n, path, k)
path.pop()
细看剪枝与不剪枝的区别仅在于for循环的终止条件改变了
当进入for循环时,已知的是path的长度,这是已经记录在案的内容,题目要求的长度是k,那此时,还需要k-len(path)个数,当i=n-(k-len(path))+1,从当前i的值算起,到n为止,正常还有k-len(path)个数,但当i=n-(k-len(path))+2时,从当前i算起,到n,只剩下k-len(path)-1个数,就算把他们全扔进path中,也达到不了len(path)==k这个终止条件,这意味着这条路径绝对返还不了有用的结果,那就提前裁剪掉。