回溯问题其实就像是多叉树的遍历,有几个关键的点。一个是候选点列表的改变,一个是最终保存的结果。一个是路径。回溯的关键在于回退到上一步的状态。
那么我们以全排列为切入点,理解一下回溯思想。
我们以python为例:
既然是多叉树的遍历,那么我们可以逐步缩小候选列表,也可以用剪枝的方式实现。剪枝的方式简单一些:
剪枝:
def permutation(nums):
res = [] # 最终结果
# 路径: track
# 候选列表: 动态变化 nums
def backtrack(choose, track):
if len(choose) == len(track):
res.append(track[:])
return
for i in choose:
if i in track:
continue
track.append(i)
backtrack(choose, track)
track.pop()
backtrack(nums, [])
return res
if __name__ == "__main__":
nums=[1, 2, 3]
res = permutation(nums)
print(res)
选择列表动态变化:
def permutation(nums):
res = [] # 最终结果
# 路径: track
# 候选列表: 动态变化 nums
def backtrack(choose, track):
if choose == []:
res.append(track[:])
return
for i in choose:
track.append(i)
choose_copy = choose[:] # 为什么这里要拷贝一下,然后再remove呢
# 因为不拷贝的化,回退的时候,候选框列表无法变回来,这也是非常难受的一个点。
choose_copy.remove(i)
backtrack(choose_copy, track)
track.pop()
backtrack(nums, [])
return res
if __name__ == "__main__":
nums=[1, 2, 3]
res = permutation(nums)
print(res)