[Leetcode][第216题][JAVA][数组之和3][回溯]

【问题描述】[中等]

【解答思路】

回溯 剪树枝 当和超过n 或 个数超过k

1. 正向求和 优化前

class Solution {
    public List<List<Integer>>  ans = new LinkedList();
    public List<List<Integer>> combinationSum3(int k, int n) {
        dfs(0,k,0,n,1,new LinkedList<Integer>());

        return ans;
    }

    private  void dfs(int depth,int k,int sum,int n,int num,List<Integer> list){

        if(depth>k){
            return;
        }
        if(sum>n){
            return;
        }

        if(depth ==k && sum==n){
            ans.add(new LinkedList<Integer>(list));
            return;
        }
       


        for(int i = num;i<=9;i++){
            sum+=i;
            list.add(i);
            dfs(depth+1,k,sum,n,i+1,list);
            list.remove(list.size()-1);
            sum-=i;

        }
    }
}

2. 逆向减等0 优化后

    public List<List<Integer>> combinationSum3(int k, int n) {
        List<List<Integer>> res = new ArrayList<>();
        dfs(res, new ArrayList<>(), k, 1, n);
        return res;
    }

    private void dfs(List<List<Integer>> res, List<Integer> list, int k, int start, int n) {
        //终止条件，如果满足这个条件，再往下找也没什么意义了
        if (list.size() == k || n <= 0) {
            //如果找到一组合适的就把他加入到集合list中
            if (list.size() == k && n == 0)
                res.add(new ArrayList<>(list));
            return;
        }
        for (int i = start; i <= 9; i++) {
            //选择当前值
            list.add(i);
            //递归
            dfs(res, list, k, i + 1, n - i);
            //撤销选择
            list.remove(list.size() - 1);
        }
    }

【总结】

1. 逆向可以减少 传参数量

2.在函数中调用dfs时，直接初始化变量，不用另外创建新的变量

3.回溯算法相关题目

[Leedcode][JAVA][第46题][全排列][回溯算法]
[Leetcode][第81题][JAVA][N皇后问题][回溯算法]
[Leetcode][第60题][JAVA][第k个排列][回溯][DFS][剪枝]

转载