代码随想录算法训练营第23天 | 回溯进阶

2024年7月25日

题39. 组合总和
由于每个元素可以用多次，要想到在每次递归里还要循环即可。
代码首先给各个候选排序，从小到大依次提高门槛，每次回溯就提高index。

class Solution {

    List<List<Integer>> res;
    List<Integer> path;
    int target;
    int[] candidates;

    public List<List<Integer>> combinationSum(int[] candidates, int target) {
        Arrays.sort(candidates);
        this.target = target;
        this.candidates = candidates;
        res = new ArrayList<>();
        path = new ArrayList<>();
        for(int i=0;i<candidates.length;i++){
            path.add(candidates[i]);
            int index = i;
            backtracking(index);
            path.clear();
        }
        return res;
    }

    public void backtracking(int index){
        int sum = path.stream().mapToInt(Integer::intValue).sum();
        if(sum==target){
            res.add(new ArrayList<>(path));
            return;
        }
        for(int i=index;i<candidates.length;i++){
            if(sum+candidates[i]>target){
                return;
            }else{
                path.add(candidates[i]);
                backtracking(index);
                path.removeLast();
                index+=1;
            }
        }
    }
}