回溯算法
1 介绍
回溯算法是一种择优搜索算法,通俗来说是一种试探算法,遍历每一种可能,满足要求则记录下来,不满足则回退到上一种可能。
用一个简单的例子来解释回溯算法,我们且需要去往某个地方,路上有许多分岔路口,在面临分叉路口的选择时,我们先尝试其中一个,走不通退回到分叉路,选择另外一条路。一直到走到目的地为止。
2 相关题目
2.1 全排列
有一个非常经典回溯算法题目,全排列。
leetcode 46
给定一个 没有重复 数字的序列,返回其所有可能的全排列。
输入: [1,2,3]
输出:
[
[1,2,3],
[1,3,2],
[2,1,3],
[2,3,1],
[3,1,2],
[3,2,1]
]
思路: 我们需要列举出所有可能的排列组合,这正好符合回溯算法的特点,我们需要试探所有的可能性,只是在这里,我们不是需要选择一个最优可能,而是记录所有可能。
针对当前这道题,有一个很清晰的思路过程:
1、在我们选择第一个数时,有3个可能,1,2,3.假设我们选定1。
2、到第二个数时有两个选择2,3,假设我们选定2。
3、到第三个数已经没得选择了,只能选择3。这时我们的得到一个答案1,2,3;
4、接着按照回溯算法的思路,我们需要回退,将3从答案中剔除,选择其它数字代替3。因为没有其它数可以代替3了,所有继续回退。
5、回退到第二个数,还记得我们在第二个数是有两个选择2,3.这时我们不再选择2,而是选择3。
6、到第三个数,我们就可以选择2,得到另外一个答案,1,3,2;我们可以按照这样的思路画出示意图。
考虑到我们需要回退,所以在代码实现时采用递归是非常合适的。
实现思路: 我们需要使用一个数组来记录当前元素是否已经被使用,used[i]表示nums[i]已被使用,不要再重复使用了。遍历到没被使用的元素,将其加入到答案数组中,当答案数组长度到达原数组长度,输出答案并且回退,寻找其它答案。
代码实现
class Solution {
List<List<Integer>> res;
List<Integer> ans;
int n;
int[] used;
public List<List<Integer>> permute(int[] nums) {
res = new ArrayList<>();
ans = new ArrayList<>();
n = nums.length;
used = new int[n];
dfs(nums,0);
return res;
}
public void dfs(int[] nums, int index){
if(index ==