标签：int res sum dfs 无标题 path size

思路

原题链接

1. 要求找出相加之和为n 的k个数字；比如和为7的3个数 1 2 4 。其实就是在1-9内取k个数，并且k个数的和为n
2. 与之前的leetcode77不同的是，此处对和值进行了限制
3. 需要定义一个全局变量sum，在for循环开始之后，不仅要将 i 加入到path中，还要将sum += i
4. 在dfs结束之后，不仅要执行removeLast操作，还要将sum -= i
5. 优化：进行剪枝，可以在sum>n 的时候，进行剪枝，已经在for循环寻找上界的时候进行剪枝
6. 在寻找上界的时候，思路与leetcode77相同：
7. 只不过此处是1-9之间的数字进行选择
   // 要选择的元素个数 = k - path.size()，整理得到：
   // 搜索起点的上界 = 9 - (k - path.size()) + 1
   // 把 i <= n 改成 i <= 9 - (k - path.size()) + 1 ：

class Solution {
    int sum = 0;
    public List<List<Integer>> combinationSum3(int k, int n) {
        List<List<Integer>> res = new ArrayList<>();
        if(k <= 0 || n <= 0 ) return res;
        Deque<Integer> path = new ArrayDeque<>();
        dfs(k, n, 1, path, res);
        return res; 
    }
    private void dfs(int k, int n, int begin, Deque<Integer> path, List<List<Integer>> res){
        //终止条件？ 当什么时候，将path加入到res中？应该是长度为k，并且sum为n
        // 减枝
		if (sum > n) {
			return;
		}
        if(path.size() == k && sum == n){
            res.add(new ArrayList<>(path));
            return;
        }
        // 减枝 9 - (k - path.size()) + 1
        for(int i = begin; i <= 9 - (k - path.size()) + 1; i++){
        // 未减枝
        // for(int i = begin; i <= 9; i++){
            path.addLast(i);
            sum += i;
            dfs(k, n, i + 1, path, res);
            path.removeLast();
            sum -= i;
        }
    }
}

简洁版

class Solution {
    int sum = 0;
    public List<List<Integer>> combinationSum3(int k, int n) {
        List<List<Integer>> res = new ArrayList<>();
        if(k <= 0 || n <= 0 ) return res;
        Deque<Integer> path = new ArrayDeque<>();
        dfs(k, n, 1, path, res);
        return res; 
    }
    private void dfs(int k, int n, int begin, Deque<Integer> path, List<List<Integer>> res){
        // 减枝
		if (sum > n) {
			return;
		}
        if(path.size() == k && sum == n){
            res.add(new ArrayList<>(path));
            return;
        }
        // 减枝 9 - (k - path.size()) + 1
        for(int i = begin; i <= 9 - (k - path.size()) + 1; i++){
            path.addLast(i);
            sum += i;
            dfs(k, n, i + 1, path, res);
            path.removeLast();
            sum -= i;
        }
    }
}

标签：int,res,sum,dfs,无标题,path,size
来源： https://blog.csdn.net/Hong__Yan/article/details/123108900

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。