回溯法
leetcode例题39、40、46、47、77、78、90
46、给定一个 没有重复 数字的序列,返回其所有可能的全排列
public List<List<Integer>> permute(int[] nums) {
List<List<Integer>> list = new ArrayList<>();
backtract(nums,list,new ArrayList<>(),0);
return list;
}
public void backtract(int[] nums,List<List<Integer>> list,List<Integer> tempList,int start){
if(tempList.size() == nums.length){
list.add(new ArrayList<>(tempList));
return;
}
for(int i=start;i<nums.length;i++){
if(tempList.contains(nums[i])){
continue;
}
tempList.add(nums[i]);
backtract(nums,list,tempList,0);
tempList.remove(tempList.size()-1);
}
}
问题1 为什么 list.add(new ArrayList<>(tempList))而不是直接list.add(tempList)
变量 tempList 所指向的列表在深度优先遍历的过程中只有一份 ,深度优先遍历完成以后,回到了根结点,成为空列表。
在 Java 中,参数传递是值传递,对象类型变量在传参的过程中,复制的是变量的地址。这些地址被添加到 list变量,但实际上指向的是同一块内存地址,因此我们会看到空的列表对象。解决的方法很简单,list.add(new ArrayList<>(tempList))这里做一次拷贝即可。
47、给定一个可包含重复数字的序列 nums ,按任意顺序 返回所有不重复的全排列。
class Solution {
public List<List<Integer>> permuteUnique(int[] nums) {
List<List<Integer>> list = new ArrayList<>();
//定义一个flag数组,用于记录形成的序列中某个元素是否已经在列表中
boolean[] flag = new boolean[nums.length];
//注意要将数组排序
Arrays.sort(nums);
backtract(nums,list,new ArrayList<>(),0,flag);
return list;
}
public void backtract(int[] nums,List<List<Integer>> list,List<Integer> tempList,int start,boolean[] flag){
if(tempList.size() == nums.length){
list.add(new ArrayList<>(tempList));
return;
}
for(int i=start;i<nums.length;i++){
if(i>0 && nums[i] == nums[i-1] && flag[i-1] == false){
continue;
}
if(!(flag[i])){
tempList.add(nums[i]);
flag[i] = true;
backtract(nums,list,tempList,0,flag);
flag[i] = false;
tempList.remove(tempList.size()-1);
}
}
}
}
问题2
if(i>0 && nums[i] == nums[i-1] && flag[i-1] == false){
continue;
}
此处用于判断从当前位置开始回溯的元素是否与之前已经完成回溯的相同元素重复。降低了时间复杂度。
77、给定两个整数 n 和 k,返回 1 … n 中所有可能的 k 个数的组合。
class Solution {
public List<List<Integer>> combine(int n, int k) {
List<List<Integer>> list = new ArrayList<>();
backtract(n,list,new ArrayList<Integer>(),k,1);
return list;
}
public void backtract(int n,List<List<Integer>> list,List<Integer> tempList,int k,int start){
if(tempList.size() == k){
list.add(new ArrayList<>(tempList));
return;
}
//注意此处剪枝的方式
for(int i=start;i<=n-k+tempList.size()+1;i++){
tempList.add(i);
backtract(n,list,tempList,k,i+1);
tempList.remove(tempList.size()-1);
}
}
}
问题3 使用回溯法之后要对代码进行剪枝,从而降低算法的时间复杂度