【LeetCode】39. 组合总和

39. 组合总和

知识点：递归；回溯；组合；剪枝

题目描述

给定一个无重复元素的正整数数组 candidates 和一个正整数 target ，找出 candidates 中所有可以使数字和为目标数 target 的唯一组合。

candidates 中的数字可以无限制重复被选取。如果至少一个所选数字数量不同，则两种组合是唯一的。

对于给定的输入，保证和为 target 的唯一组合数少于 150 个。

示例

输入: candidates = [2,3,6,7], target = 7

输出: [[7],[2,2,3]]

输入: candidates = [2,3,5], target = 8

输出: [[2,2,2,2],[2,3,3],[3,5]]   

输入: candidates = [2], target = 1

输出: []  

输入: candidates = [1], target = 1

输出: [[1]]  

输入: candidates = [1], target = 2

输出: [[1,1]]

解法一：回溯

回溯算法的模板：

result = []   //结果集

def backtrack(路径, 选择列表):

    if 满足结束条件:

        result.add(路径)  //把已经做出的选择添加到结果集；

        return  //一般的回溯函数返回值都是空；

    for 选择 in 选择列表: //其实每个题的不同很大程度上体现在选择列表上，要注意这个列表的更新，

    //比如可能是搜索起点和重点，比如可能是已经达到某个条件，比如可能已经选过了不能再选；

        做选择  //把新的选择添加到路径里；路径.add(选择)

        backtrack(路径, 选择列表) //递归；

        撤销选择  //回溯的过程；路径.remove(选择)

核心就是for循环里的递归，在递归之前做选择，在递归之后撤销选择；

对于本题，有两点和77题组合不一样：

此题可以重复选取选过的元素，所以选择列表的搜索起点不用i+1，仍然是i。
此题没有像之前的题明确给出递归的层数，但是给了target，所以如果相加>target,那就证明到头了；

我们换个角度重新画这个图，和77题有点差距，理解的更全面一点。其实这就是一个横向循环和纵向的递归，横向循环做出不同的选择，纵向在不同的选择基础上做下一步选择。

class Solution {

    public List<List<Integer>> combinationSum(int[] candidates, int target) {

        List<List<Integer>> res = new ArrayList<>();

        Stack<Integer> path = new Stack<>();

        backtrack(candidates, target, 0, 0, res, path);

        return res;

    }

    private void backtrack(int[] candidates, int target, int sum, int begin, List<List<Integer>> res, Stack<Integer> path){

        if(sum > target){

            return;

        }

        if(sum == target){

            res.add(new ArrayList<>(path));

            return;

        }

        for(int i = begin; i < candidates.length; i++){

            //做选择；

            sum += candidates[i];

            path.push(candidates[i]);

            //递归：开始下一轮选择；

            backtrack(candidates, target, sum, i, res, path);  //不用+1，可以重复选；

            //撤销选择：回溯

            sum -= candidates[i];

            path.pop();

        }

    }

}

解法二:剪枝优化

上述程序有优化的空间，我们可以对数组先进行排序，然后如果找到了当前的sum已经等于target或大于target了，那后面的就可以直接跳过了，因为后面的元素更大，肯定更大于target。

class Solution {

    public List<List<Integer>> combinationSum(int[] candidates, int target) {

        List<List<Integer>> res = new ArrayList<>();

        Stack<Integer> path = new Stack<>();

        Arrays.sort(candidates);  //排序

        backtrack(candidates, target, 0, 0, res, path);

        return res;

    }

    private void backtrack(int[] candidates, int target, int sum, int begin, List<List<Integer>> res, Stack<Integer> path){

        if(sum == target){

            res.add(new ArrayList<>(path));

            return;

        }

        for(int i = begin; i < candidates.length && sum + candidates[i] <= target; i++){

            //剪枝：如果sum+candidates[i] > target就结束；

            //做选择；

            sum += candidates[i];

            path.push(candidates[i]);

            //递归：开始下一轮选择；

            backtrack(candidates, target, sum, i, res, path);  //不用+1，可以重复选；

            //撤销选择：回溯

            sum -= candidates[i];

            path.pop();

        }

    }

}

体会

要能够把这种决策树画出来；
在求和问题中，排序之后加上剪枝是很常见的操作，能够舍弃无关的操作；

巴特西

【LeetCode】39. 组合总和

39. 组合总和

题目描述

示例

解法一：回溯

解法二:剪枝优化

体会

相关链接

最新文章

热门文章