截止到目前我已经写了 600多道算法题,其中部分已经整理成了pdf文档,目前总共有1000多页(并且还会不断的增加),大家可以免费下载 下载链接:https://pan.baidu.com/s/1hjwK0ZeRxYGB8lIkbKuQgQ 提取码:6666
全排列也是一道经典的题,之前在讲450,什么叫回溯算法,一看就会,一写就废的时候,也提到过使用回溯算法来解决,具体细节可以看下。假设数组长度是n ,我们可以把回溯算法看做是一颗n 叉树的前序遍历,第一层有n 个子节点,第二层有n-1 个子节点…… ,来看个视频
视频链接 代码如下
public List<List<Integer>> permute(int[] nums) {
List<List<Integer>> list = new ArrayList<>();
backtrack(list, new ArrayList<>(), nums);
return list;
}
private void backtrack(List<List<Integer>> list, List<Integer> tempList, int[] nums) {
if (tempList.size() == nums.length) {
list.add(new ArrayList<>(tempList));
return;
}
for (int i = 0; i < nums.length; i++) {
if (tempList.contains(nums[i]))
continue;
tempList.add(nums[i]);
backtrack(list, tempList, nums);
}
}
我们就用数组[1,2,3] 来测试一下,看一下打印结果
[1, 2, 3]
是不是很意外,示例1 给出的是6 个结果,这里打印的是1 个,这是因为list 是引用传递,当遍历到叶子节点以后要往回走,往回走的时候必须把之前添加的值给移除了,否则会越加越多,我们来看下视频 视频链接 再来看下代码
public List<List<Integer>> permute(int[] nums) {
List<List<Integer>> list = new ArrayList<>();
backtrack(list, new ArrayList<>(), nums);
return list;
}
private void backtrack(List<List<Integer>> list, List<Integer> tempList, int[] nums) {
if (tempList.size() == nums.length) {
list.add(new ArrayList<>(tempList));
return;
}
for (int i = 0; i < nums.length; i++) {
if (tempList.contains(nums[i]))
continue;
tempList.add(nums[i]);
backtrack(list, tempList, nums);
tempList.remove(tempList.size() - 1);
}
}
我们来看一下运行结果
[1, 2, 3]
[1, 3, 2]
[2, 1, 3]
[2, 3, 1]
[3, 1, 2]
[3, 2, 1]
这题使用回溯算法的还一种解决方式就是交换,比如我们先选择第一个数字,然后和后面的所有数字都交换一遍,这样全排列的第一位就确定了。然后第二个数字在和后面的所有数字交换一遍,这样全排列的第二位数字也确定了……,一直继续下去,直到最后一个数字不能交换为止,这里画个图来看一下
来看下代码
public List<List<Integer>> permute(int[] nums) {
List<List<Integer>> res = new ArrayList<>();
backtrack(nums, 0, res);
return res;
}
public void backtrack(int[] nums, int index, List<List<Integer>> res) {
if (index == nums.length - 1) {
List<Integer> tempList = new ArrayList<>();
for (int num : nums)
tempList.add(num);
res.add(tempList);
return;
}
for (int i = index; i < nums.length; i++) {
swap(nums, index, i);
backtrack(nums, index + 1, res);
swap(nums, index, i);
}
}
private void swap(int[] nums, int i, int j) {
if (i != j) {
nums[i] ^= nums[j];
nums[j] ^= nums[i];
nums[i] ^= nums[j];
}
}
递归解决
我们来思考这样一个问题,假如数组[1,2,3] ,我们知道了[2,3] 的全排列结果,只需要把1 放到这些全排列的所有位置,即是数组[1,2,3] 的全排列,画个图来看一下
public List<List<Integer>> permute(int[] nums) {
return helper(nums, 0);
}
private List<List<Integer>> helper(int[] nums, int index) {
List<List<Integer>> res = new ArrayList<>();
if (index == nums.length - 1) {
List<Integer> temp = new ArrayList<>();
temp.add(nums[index]);
res.add(temp);
return res;
}
List<List<Integer>> subList = helper(nums, index + 1);
int count = subList.get(0).size();
for (int i = 0, size = subList.size(); i < size; i++) {
for (int j = 0; j <= count; j++) {
List<Integer> temp = new ArrayList<>(subList.get(i));
temp.add(j, nums[index]);
res.add(temp);
}
}
return res;
}
|