代码随想录|回溯算法 77. 组合,216.组合总和 III,17.电话号码的字母组合
python
一、77. 组合
内容
77. 组合
1.核心代码
代码如下(示例):
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() # 回溯,撤销处理的节点
result.append(path[:])是浅拷贝。否则,因为通过append把path添加到result,是引用对象,所以后续path改变,result也会改变。而增加了[:]后,后续修改不会影响前面的引用了
2.输入输出
class Solution:
def combine(self,n,k):
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()
if __name__=="__main__":
n=int(input())
k=int(input())
solution=Solution()
result=solution.combine(n,k)
print(result)
3.问题
combine 方法调用了 backtraversal 方法,而 backtraversal 方法是定义在类内的,因此需要通过 self 进行调用。
backtraversal 方法的签名中缺少 self。没有 self 的话,方法无法正确绑定到实例上,也无法访问实例变量和方法。
二、216. 组合总和III
1.代码
class Solution:
def combinationSum3(self, k: int, n: int) :
result = []
self.back(n, k, 0, 1, [], result)
return result
def back(self, n, k, sum, start, path, result):
if sum > n:
return
if len(path) == k:
if n == sum:
result.append(path[:])
return
for i in range(start, 10):
sum = sum + i
path.append(i)
self.back(n, k, sum, i + 1, path, result)
sum = sum - i
path.pop()
if __name__=="__main__":
k=int(input())
n=int(input())
solution=Solution()
result=solution.combinationSum3(k,n)
print(result)
上一题的稍微变式
2.问题
终止条件通过多层if进行多限制
有append有pop,回溯!
三、17. 电话号码的字母组合
1.代码
代码如下(示例):
class Solution:
def __init__(self):
self.letterMap = [
"",
"",
"abc",
"def",
"ghi",
"jkl",
"mno",
"pqrs",
"tuv",
"wxyz"
]
self.result = []
self.s =""
def backtracking(self,digits,index):
if index == len(digits):
self.result.append(self.s)
return
digit = int(digits[index])
letters = self.letterMap[digit]
for i in range(len(letters)):
self.s +=letters[i]
self.backtracking(digits,index+1)
self.s = self.s[:-1]
def letterCombinations(self,digits):
if len(digits) ==0:
return self.result
self.backtracking(digits,0)
return self.result
if __name__=="__main__":
digits=str(input())
solution=Solution()
result = solution.letterCombinations(digits)
print(result)
映射问题——map
输入输出简单,不用看
这道题只看了代码,没有自己写
总结
输入输出