leetcode算法基础 第十天 递归/回溯

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;
        }
    }
}
上一篇:10.02lcode


下一篇:深度优先搜索实现回溯算法