leetcode练习
40. 组合总和 II
题目
解决数组里有重复元素的情况,就将数组排序,然后记录上次一次选择的元素,并在同层回溯中,不重复选择该元素,保证在每层回溯中,相同元素只会被选择一次。
List<List<Integer>> re = new ArrayList<>();
LinkedList<Integer> path = new LinkedList<>();
public List<List<Integer>> combinationSum2(int[] candidates, int target) {
Arrays.sort(candidates); //解决重复[1,2,5,1]会出现[1,2,5],[2,5,1]这种情况需要排序
backtracking(candidates, target, 0, 0);
return re;
}
//使用path.stream().mapToInt(Integer:intValue).sum()时间复杂度会增加
public void backtracking(int[] candidates, int target, int startIndex, int sum){
if (sum == target){
re.add(new ArrayList<>(path));
return;
}
int last = Integer.MAX_VALUE;
for (int i = startIndex; i < candidates.length; i++){
if (last == candidates[i]) continue;
if (candidates[i] > target) continue;
if (candidates[i] + sum > target) continue;
last = candidates[i];
sum += candidates[i];
path.add(candidates[i]);
backtracking(candidates, target, i + 1, sum);
path.removeLast();
sum -= candidates[i]; //记得回退
}
}
47. 全排列 II
题目
这道题是选择一个具有重复元素数组的全排列,一般的回溯模板要有startIndex,因为全排列所以每一个没被选择的元素都有可能被使用,所以设置一个visit数组,使用过的元素不能再次选择。同时为了解决重复元素带来的影响,同样设置一个last来记录上一次选择的元素,同层回溯不能选择相同元素。
List<List<Integer>> re = new ArrayList<>();
LinkedList<Integer> path = new LinkedList<>();
public List<List<Integer>> permuteUnique(int[] nums) {
int[] visit = new int[nums.length];
Arrays.sort(nums);
backtracking(visit, nums);
return re;
}
public void backtracking(int[] visit, int nums[]){
if (path.size() == nums.length){
re.add(new ArrayList<>(path));
return;
}
int last = Integer.MAX_VALUE; //记录上一次使用的元素,需要数组有序
for (int i = 0; i < nums.length; i++){
if (nums[i] == last) continue; //同层内的相同元素不能重复使用
if (visit[i] == 0){
last = nums[i];
path.add(nums[i]);
visit[i] = 1;
backtracking(visit, nums);
path.removeLast();
visit[i] = 0;
}
}
}