给定一个不含重复数字的数组 nums
,返回其 所有可能的全排列 。你可以 按任意顺序 返回答案。
归类为回溯题,回溯题想到用n叉树来进行解决,首先明确:结果只保存在叶子节点,非叶子节点不是需要的结果。并且不需要剪枝,因为要从头开始并避开当前节点,树的每层有n - k 个分支(k为层数)。
先上代码,然后我讲一下自己的思路。
import java.util.ArrayList;
import java.util.List;
public class no_46 {
public static void main(String[] args) {
int[] nums = {1, 2, 3};
List<List<Integer>> permute = permute(nums);
for (List<Integer> per : permute) {
System.out.println(per);
}
}
public static List<List<Integer>> permute(int[] nums) {
List<List<Integer>> re = new ArrayList<>();
if (nums == null || nums.length == 0) return re;
boolean[] used = new boolean[nums.length];
List<Integer> permutation = new ArrayList<>();
backTrack(nums, used, permutation, re);
return re;
}
private static void backTrack(int[] nums, boolean[] used, List<Integer> permutation, List<List<Integer>> re) {
if (permutation.size() == nums.length) {
re.add(new ArrayList<>(permutation));
return;
}
for (int i = 0; i < nums.length; i++) {
if (used[i]) continue;
used[i] = true;
permutation.add(nums[i]);
backTrack(nums, used, permutation, re);
permutation.remove(permutation.size() - 1);
used[i] = false;
}
}
}
首先回溯要明确一点就是,使用递归是基本,所以递归前要确定使用哪些变量,这里在调用BackTrack前,建立List<List<Integer>> re,来存放最终结果,建立List<Integer> permutation来存放当前叶子节点的结果。本题不实用startIndex而使用一个boolean数组来标志当前节点是否已经被加到permutation,也就是上面我提到的这个结果不需要剪枝,而是一个满树,即倒数第二层的每个节点都有一样数量的叶子节点。
进入backTrack后,如果permutation的size和nums.length相等,那就直接将结果加入到re中,然后return,代表加入了一个叶子节点。
如果不是,那么要扫一遍nums数组,如果当前节点没被使用,那么将当前节点加入permutation,然后调用backTrack遍历剩余节点,在加入了一个叶子节点到最终re的结果后,剔除permutation最后一个数,寻找别的组合。