算法练习第22天|39. 组合总和、40.组合总和II

39. 组合总和

39. 组合总和 - 力扣(LeetCode)icon-default.png?t=N7T8https://leetcode.cn/problems/combination-sum/description/

题目描述:

给你一个 无重复元素 的整数数组 candidates 和一个目标整数 target ,找出 candidates 中可以使数字和为目标数 target 的 所有 不同组合 ,并以列表形式返回。你可以按 任意顺序 返回这些组合。

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

对于给定的输入,保证和为 target 的不同组合数少于 150 个。

示例 1:

输入:candidates = [2,3,6,7], target = 7
输出:[[2,2,3],[7]]
解释:
2 和 3 可以形成一组候选,2 + 2 + 3 = 7 。注意 2 可以使用多次。
7 也是一个候选, 7 = 7 。
仅有这两种组合。

示例 2:

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

示例 3:

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

提示:

  • 1 <= candidates.length <= 30
  • 2 <= candidates[i] <= 40
  • candidates 的所有元素 互不相同
  • 1 <= target <= 40

 思路分析:

由于所有candidates的数>=1,所以不必担心有0的问题。另外,由于元素可以多次使用,所以其搜索过程对应的树形结构如下所示:

注意图中叶子节点的返回条件,因为本题没有组合数量要求,仅仅是总和的限制,所以递归没有层数的限制,只要选取的元素总和超过target,就返回! 另外,注意蓝色框中文字,因为要求的是满足条件的

下面我们结合上述的树形结构图,按照回溯三部曲来尝试书写代码:

  • 回溯第一步:确认回溯函数的参数和返回值。

这里依然是定义两个全局变量,二维数组result存放结果集,数组path存放符合条件的结果。(这两个变量可以作为函数参数传入) 

老规矩,返回值类型void。参数除了题目所给的数组candidates以及目标和target之外,为了方便理解,我们算法练习第21天|216.组合总和|||、17.电话号码的字母组合-CSDN博客中的216.组合总和|||的解法一样,添加了目前路径上所遍历的元素的和sum以及开始遍历的位置startIndex。

可能有人有人会有疑问,为什么在17.电话号码的字母组合这一题中没有写startIndex?对于组合问题,什么时候需要startIndex呢?下面给出startIndex的使用场景:

如果是从一个集合中求组合的话,就需要startIndex,例如算法练习第20天|回溯算法 77.组合问题 257. 二叉树的所有路径-CSDN博客中的77.组合问题,以及上述的216.组合总和|||。

如果是多个集合取组合,各个集合之间相互不影响,那么就不用startIndex,例如:上面的17.电话号码的字母组合。

所以本题要求是从同一个集合中求组合,所以要用到startIndex。至于具体怎么用,下面代码中在细说。

回溯函数大致长这样:

vector<vector<int>> result;
vector<int> path;
//回溯第一步,确定回溯函数的参数和返回类型
void backTracking(vector<int>& candidates, int target, int sum, int startIndex){}
  • 回溯函数第二步:确认回溯终止条件 

从下面的树形结构可知, 回溯的终止条件只有 sum== targeth和sum > target两种。如果是sum<target,则说明还有继续寻找并添加元素的可能,所以这种要执行的是单层回溯的逻辑。

所以回溯终止条件长这样:

//回溯第二步,确定回溯终止条件
if(sum == target){
     result.push_back(path);
     return;
}
else if(sum > target){
     return;
}
  •  回溯第三步:确认单层搜索逻辑

能到达这一步,表明sum<target,需要继续进行元素搜索。因此,要做的操作有处理节点(sum加上当前节点,path中记录当前节点),然后就该执行递归了,接着是回溯。代码如下:

//回溯第三步,确认当层回溯逻辑,并回溯
for(int i = startIndex; i < candidates.size(); i++){
      sum += candidates[i];  //处理节点
      path.push_back(candidates[i]);

      backTracking(candidates, target, sum, i);  //递归

      sum -= candidates[i]; //回溯
      path.pop_back();

}

这里我们用到了startIndex,注意我们在递归时的给最后一个参数传的是i,程序初始时传的startIndex肯定为0,所以i最开始为0。但是搜索完最左侧的大分支后,该记录的结果记录下来,该回溯的回溯,i++变成了1,结合下图,为了避免上面所说的【2,3】和【3,2】的组合重复,所以取数时变成了从【3,5】中取数,这就导致startIndex也需要有更新。那么更新为多少时何时呢?观察下图可知当startIndex和此时的i一样时就满足了从【3,5】中取数的条件,于是我们在递归时把i作为了第四个参数的实参。

整体代码:

 整体代码的书写如下:

class Solution {
public:
    vector<vector<int>> result;
    vector<int> path;
    //回溯第一步,确定回溯函数的参数和返回类型
    void backTracking(vector<int>& candidates, int target, int sum, int startIndex){
        //回溯第二步,确定回溯终止条件
        if(sum == target){
            result.push_back(path);
            return;
        }
        else if(sum > target){
            return;
        }

        //回溯第三步,确认当层回溯逻辑,并回溯
        for(int i = startIndex; i < candidates.size(); i++){
            sum += candidates[i];  //处理节点
            path.push_back(candidates[i]);

            backTracking(candidates, target, sum, i);  //递归

            sum -= candidates[i]; //回溯
            path.pop_back();

        }
    }

    vector<vector<int>> combinationSum(vector<int>& candidates, int target) {
        backTracking(candidates,target,0,0);
        return result; 
    }
};

40.组合总和II 

40. 组合总和 II - 力扣(LeetCode)icon-default.png?t=N7T8https://leetcode.cn/problems/combination-sum-ii/description/

题目描述:

给定一个候选人编号的集合 candidates 和一个目标数 target ,找出 candidates 中所有可以使数字和为 target 的组合。

candidates 中的每个数字在每个组合中只能使用 一次 。

注意:解集不能包含重复的组合。 

示例 1:

输入: candidates = [10,1,2,7,6,1,5] target = 8输出:
[
[1,1,6],
[1,2,5],
[1,7],
[2,6]
]

示例 2:

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

提示:

  • 1 <= candidates.length <= 100
  • 1 <= candidates[i] <= 50
  • 1 <= target <= 30

思路分析:

这道题目和39.组合总和 (opens new window)有两个区别:

  1. 本题candidates 中的每个数字在每个组合中只能使用一次。
  2. 本题数组candidates的元素是有重复的,而39.组合总和 (opens new window)是无重复元素的数组candidates

重复的元素则会产生重复的组合, 如线图所示。集合【1,1,2】中的1是重复的元素,这就导致在搜索遍历的过程中产生两次满足条件的【1,2】组合。所以要想办法去掉这些重复的组合。去重的一个技巧就是先对数组进行排序。比如集合是【1,2,1】,那么不排序的情况下会产生【1,2】和【2,1】两个过程组合,但他们属于同一个组合,是重复的。所以在进行功能实现前先对candidates数组进行排序。

从上图中可以看出,重复情况出现在同一树层上。而for循环决定着横向遍历的树层,依次判断重复取同一个数字的条件就是遍历的 i > startIndex && candidates[i] == candidates[i-1],如下图所示:

经过上面的分析,下面开始日常回溯三部曲。

首先,按照老规矩定义path和result两个全局变量。

  • 回溯第一步:确认回溯函数的返回值和参数。返回值void,参数除了candidates,还要有目前的和sum以及startIndex。

        按照前面39题思路中所讲的,同一个集合中求组合使用startIndex,不同的集合求组合则无使用startIndex。在递归时,元素允许重复使用则给startIndex传i,不允许重复使用则传i+1.

下面给出回溯函数的样子:

vector<int> path;
vector<vector<int>> result;

//回溯第一步,确认回溯函数的返回值与参数
void backTracking(vector<int>& candidates, int target, int sum, int startIndex){}
  • 回溯第二步:确认回溯终止条件。 

终止的情况有两种:(1)满足条件,sum==target,result记录path,return;

(2)sum>target, 则直接return。

其余的情况都应该发生在单层索搜的功能中。

因此给出第二步的代码:

        //回溯第二步,确认终止条件
        if(sum > target){
            return;
        }
        else if(sum == target){
            result.push_back(path);
            return;
        }
  •  回溯第三步:确认单层搜索

 单层搜索就发生在for循环中,为了去重,在单层搜索前应先判断元素是否重复选取了。比如图中第二个1,如果是重复选取,则直接continue,开始下一层循环。如果未重复选取,则path记录candidates【i】,sum累加求和,然后是熟悉的递归(给startIndex位置传i+1)、回溯。

代码如下:

        //回溯第三部,单层搜索逻辑
        for(int i = startIndex; i < candidates.size(); i++){
            //处理节点
            //通过i和startIndex的大小关系以及candidates的i-1和i两个元素的大小来判断是否重复
            if(i>startIndex && candidates[i-1] == candidates[i])
                continue;

            sum += candidates[i];
            path.push_back(candidates[i]);

            //递归
            backTracking(candidates, target, sum, i+1);

            //回溯
            sum -= candidates[i];
            path.pop_back();

        }

 整体代码:

完整代码如下:

class Solution {
public:
    vector<int> path;
    vector<vector<int>> result;

    //回溯第一步,确认回溯函数的返回值与参数
    void backTracking(vector<int>& candidates, int target, int sum, int startIndex){
        //回溯第二步,确认终止条件
        if(sum > target){
            return;
        }
        else if(sum == target){
            result.push_back(path);
            return;
        }

        //回溯第三部,单层搜索逻辑
        for(int i = startIndex; i < candidates.size(); i++){
            //处理节点
            //通过i和startIndex的大小关系以及candidates的i-1和i两个元素的大小来判断是否重复
            if(i>startIndex && candidates[i-1] == candidates[i])
                continue;

            sum += candidates[i];
            path.push_back(candidates[i]);

            //递归
            backTracking(candidates, target, sum, i+1);

            //回溯
            sum -= candidates[i];
            path.pop_back();

        }

    }

    vector<vector<int>> combinationSum2(vector<int>& candidates, int target) {
        sort(candidates.begin(), candidates.end());
        backTracking(candidates, target, 0, 0);
        return result;
    }
};

  • 20
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值