力扣216题组合总和Ⅲ Java版

伊俊客

已于 2022-12-22 15:24:10 修改

阅读量60

点赞数

分类专栏：力扣刷题回溯文章标签： leetcode 算法 java

于 2022-12-22 14:55:19 首次发布

本文链接：https://blog.csdn.net/weixin_44960905/article/details/128408685

版权

力扣刷题回溯专栏收录该内容

14 篇文章 0 订阅

订阅专栏

class Solution {
    // 记录最终结果
    List<List<Integer>> res = new ArrayList<>();
    // 暂存每一步
    List<Integer> paths = new ArrayList<>();

    public List<List<Integer>> combinationSum3(int k, int n) {
        backtracking(k,n,0,1);
        return res;
    }


    public void backtracking(int k,int targetSum,int sum,int startIndex){
        // 当前元素为k个且当前元素的总和达到目标结果时,收集结果
        if(paths.size() == k){
            if(sum == targetSum){
                res.add(new ArrayList<>(paths));
                return;
            }
        }
	    // 遍历1~9
        for(int i=startIndex; i<=9; i++){
            paths.add(i);
            sum +=i; 
            // 递归
            backtracking(k,targetSum,sum,i+1);
            // 回溯
            paths.remove(paths.size()-1);
            sum -=i;
        }
    }
}

运行结果：
在这里插入图片描述

剪枝优化

class Solution {
    List<List<Integer>> res = new ArrayList<>();
    List<Integer> paths = new ArrayList<>();

    public List<List<Integer>> combinationSum3(int k, int n) {
        backtracking(k,n,0,1);
        return res;
    }


    public void backtracking(int k,int targetSum,int sum,int startIndex){
        // 当前结果大于目标结果,程序结束
        if(sum > targetSum){
            return;
        }
        if(paths.size() == k){
            if(sum == targetSum){
                res.add(new ArrayList<>(paths));
                return;
            }
        }
	    // 根据集合中需要的元素个数,计算出最大下标
        for(int i=startIndex; i<=9-(k-paths.size())+1; i++){
            paths.add(i);
            sum +=i; 
            backtracking(k,targetSum,sum,i+1);
            paths.remove(paths.size()-1);
            sum -=i;
        }
    }
}