一. 回溯算法简介
回溯算法就是我们常说的DFS算法,本质上是一种暴力穷举算法。
解决一个回溯问题,实际上就是一个决策树的遍历过程。
- 路径: 也就是已经做出的选择
- 选择列表: 也就是你当前可以做的选择
- 结束条件: 也就是到达决策树底层,无法再做选择的条件。
二、回溯算法的框架
result = []
def backtrack(路径, 选择列表):
if 满足结束条件:
result.add(路径)
return
for 选择 in 选择列表:
做选择
backtrack(路径,选择列表)
撤销选择
其核心就是for循环里面的递归,在递归调用之前做选择,在递归调用之后撤销选择
- 全排列问题
List<List<Integer>> res = new LinkedList<>();
/* 主函数,输入一组不重复的数字,返回它们的全排列 */
List<List<Integer>> permute(int[] nums) {
// 记录「路径」
LinkedList<Integer> track = new LinkedList<>();
backtrack(nums, track);
return res;
}
// 路径:记录在 track 中
// 选择列表:nums 中不存在于 track 的那些元素
// 结束条件:nums 中的元素全都在 track 中出现
void backtrack(int[] nums, LinkedList<Integer> track) {
// 触发结束条件
if (track.size() == nums.length) {
res.add(new LinkedList(track));
return;
}
for (int i = 0; i < nums.length; i++) {
// 排除不合法的选择
if (track.contains(nums[i]))
continue;
// 做选择
track.add(nums[i]);
// 进入下一层决策树
backtrack(nums, track);
// 取消选择
track.removeLast();
}
}