【LeetCode】39. Combination Sum（中等难度）

最新推荐文章于 2024-11-11 20:59:17 发布

玳宸

最新推荐文章于 2024-11-11 20:59:17 发布

阅读量100

点赞数

分类专栏： LeetCode 文章标签： leetcode 回溯法

本文链接：https://blog.csdn.net/zeroheitao/article/details/120759749

版权

LeetCode 专栏收录该内容

39 篇文章 0 订阅

订阅专栏

这篇博客介绍了两种使用回溯法求解组合求和问题的方法。第一种是基本的回溯法，通过尝试所有可能的组合并回溯来找到目标和。第二种方法加入了剪枝策略，通过对输入数组进行排序，减少了不必要的回溯次数，提高了算法效率。两种方法都实现了从给定的数列中找出所有加起来等于目标值的组合。

摘要由CSDN通过智能技术生成

在这里插入图片描述

方法一回溯法（未剪枝）

就是向前列举所有情况，每个值都有直接跳过和添加两个选择，得到一个解或者走不通的时候就回溯。

class Solution {
    public List<List<Integer>> combinationSum(int[] candidates, int target) {
        List<List<Integer>> res = new ArrayList<>();
        List<Integer> combine = new ArrayList<>();
        backtrack(candidates, target, res, combine, 0);
        return res;
    }

	// target：求和数	res:最终结果		combine：已经组合的数		index：当前下标
    public void backtrack(int[] candidates, int target, List<List<Integer>> res, List<Integer> combine, int index){
        if(index == candidates.length)  return;
        if(target == 0){
            res.add(new ArrayList<Integer>(combine));
            return;
        }
        backtrack(candidates, target, res, combine, index + 1);
        if(target - candidates[index] >= 0){
            combine.add(candidates[index]);
            backtrack(candidates, target - candidates[index], res, combine, index);
            combine.remove(combine.size() - 1);
        }
    }
}

方法二回溯法（剪枝）

对 candidates 排序，以减少回溯次数。

class Solution {
    private List<List<Integer>> res = new ArrayList<>();
    public List<List<Integer>> combinationSum(int[] candidates, int target) {
        List<Integer> combine = new ArrayList<>();
        Arrays.sort(candidates);
        backtrack(candidates, target, combine, 0, 0);
        return res;
    }
    public void backtrack(int[] candidates, int target, List<Integer> combine, int index, int sum){
        if(index == candidates.length)  return;
        if(target == sum){
            res.add(new ArrayList<Integer>(combine));
            return;
        }
        for(int i = index; i < candidates.length; i++){
            int rs = candidates[i] + sum;
            if(rs <= target){
                combine.add(candidates[i]);
                backtrack(candidates, target, combine, i, rs);
                combine.remove(combine.size() - 1);
            }else{
                break;
            }
        }
    }
}