leetcode：216. 组合总和 III

OceanStar的学习笔记

已于 2022-08-24 20:26:16 修改

阅读量234

点赞数

分类专栏：算法与数据结构文章标签： leetcode 算法深度优先

于 2022-05-16 16:27:32 首次发布

原文链接：https://leetcode.cn/problems/combination-sum-iii/solution/dai-ma-sui-xiang-lu-dai-ni-xue-tou-hui-s-petp/

版权

算法与数据结构专栏收录该内容

1121 篇文章 80 订阅

订阅专栏

题目来源

leetcode：combination-sum-iii

题目描述

在这里插入图片描述

题目解析

回溯

leetcode：77. 组合中要求：从集合[1…n]中选出k个数。

本题比leetcode：77. 组合多了一个限制，和为n，即：从集合[1…9]中选出k个数，令其和为n。每个树只能用一次。

本题k相当于了树的深度，9（因为整个集合就是9个数）就是树的宽度。

例如 k = 2，n = 4的话，就是在集合[1,2,3,4,5,6,7,8,9]中求 k（个数） = 2, n（和） = 4的组合。

选取过程如图：
在这里插入图片描述

三部曲：

（1）确定递归函数的参数：

需要两个全局遍历

vector<vector<int>> result; // 存放结果集
vector<int> path; // 符合条件的结果

还需要如下参数：

targetSum(int)：目标和，也就是题目中的n
k(int)：就是要求选出几个数
sum(int)：已经收集到的元素的总和，也就是path中元素的总和
startIndex(int)：下一层for循环搜索的起始位置

代码如下：

vector<vector<int>> result;
vector<int> path;
void backtracking(int targetSum, int k, int sum, int startIndex)

其实这里sum这个参数也可以省略，每次targetSum减去选取的元素数值，然后判断如果targetSum为0了，说明收集到符合条件的结果了

（2）终止条件

k限制了树的深度，因为就取k个元素，树再往下深了没有意义，所以，如果path.size()和k相等了，就终止
如果此时path里收集到的元素和sum和targetSum相同了，就放入最终结果集中

if (path.size() == k) {
    if (sum == targetSum) result.push_back(path);
    return; // 如果path.size() == k 但sum != targetSum 直接返回
}

（4）单层搜索过程

集合固定就是9个数，所以for循环固定为i <= 9

for (int i = startIndex; i <= 9; i++) {
    sum += i;
    path.push_back(i);
    backtracking(targetSum, k, sum, i + 1); // 注意i+1调整startIndex
    sum -= i; // 回溯
    path.pop_back(); // 回溯
}

整体代码如下：

class Solution {
private:
    vector<vector<int>> result; // 存放结果集
    vector<int> path; // 符合条件的结果
    // targetSum：目标和，也就是题目中的n。
    // k：题目中要求k个数的集合。
    // sum：已经收集的元素的总和，也就是path里元素的总和。
    // startIndex：下一层for循环搜索的起始位置。
    void backtracking(int targetSum, int k, int sum, int startIndex) {
        if (path.size() == k) {
            if (sum == targetSum) result.push_back(path);
            return; // 如果path.size() == k 但sum != targetSum 直接返回
        }
        for (int i = startIndex; i <= 9; i++) {
            sum += i; // 处理
            path.push_back(i); // 处理
            backtracking(targetSum, k, sum, i + 1); // 注意i+1调整startIndex
            sum -= i; // 回溯
            path.pop_back(); // 回溯
        }
    }

public:
    vector<vector<int>> combinationSum3(int k, int n) {
        backtracking(n, k, 0, 1);
        return result;
    }
};

（5）剪枝

已选元素总和如果已经大于n（图中数值为4）了，那么往后遍历就没有意义了，直接剪掉。那么剪枝的地方一定是在递归终止的地方剪，剪枝代码如下：

if (sum > targetSum) { // 剪枝操作
    return;
}

for循环的范围也可以剪枝，i <= 9 - (k - path.size()) + 1就可以了。

在这里插入图片描述
最后C++代码如下：

class Solution {
private:
    vector<vector<int>> result; // 存放结果集
    vector<int> path; // 符合条件的结果
    void backtracking(int targetSum, int k, int sum, int startIndex) {
        if (sum > targetSum) { // 剪枝操作
            return; // 如果path.size() == k 但sum != targetSum 直接返回
        }
        if (path.size() == k) {
            if (sum == targetSum) result.push_back(path);
            return;
        }
        for (int i = startIndex; i <= 9 - (k - path.size()) + 1; i++) { // 剪枝
            sum += i; // 处理
            path.push_back(i); // 处理
            backtracking(targetSum, k, sum, i + 1); // 注意i+1调整startIndex
            sum -= i; // 回溯
            path.pop_back(); // 回溯
        }
    }

public:
    vector<vector<int>> combinationSum3(int k, int n) {
        backtracking(n, k, 0, 1);
        return result;
    }
};

回溯法抽象为树形结构后，其遍历过程就是：for循环横向遍历，递归纵向遍历，回溯不断调整结果集。

思路二

class Solution {
private:
    vector<vector<int>> ans; // 最终结果集
    vector<int> cur; // 当前结果集

    /**
     * 
    * u: 当前遍历到的数字
    * sum: 当前结果集的总和
   */
    void backtracking(int n, int k, int sum, int u) {
        if(sum == n && cur.size() == k){
            ans.push_back(cur);
            return;
        }
        
        if(u == 10 || sum > n || cur.size() > k){
            return;
        }

        // 使用数字 u
        cur.push_back(u);
        backtracking(n, k, sum + u, u + 1);
        // 进行回溯
        cur.pop_back();
        // 不使用数字 u
        backtracking(n, k, sum, u + 1);
    }

public:
    vector<vector<int>> combinationSum3(int k, int n) {
        backtracking(n, k, 0, 1);
        return ans;
    }
};

小结

「回溯算法」从算法定义上来说，不一定要用 DFS 实现，但通常结合 DFS 来做，难度是最低的。

「回溯算法」根据当前决策有多少中选择，对应了两套模板

每一次独立的决策值对应选择和不选两种情况（对应到这类模板的题目有：40. 组合总和 II ）
- 确定接收回溯过程的base case
- 遍历每个位置，对每个位置进行决策（做选择 -> 递归 -> 撤销选择）

void dfs(当前位置, 路径(当前结果), 结果集) {
    if (当前位置 == 结束位置) {
        结果集.add(路径);
        return;
    }
        
    选择当前位置;    
    dfs(下一位置, 路径(当前结果), 结果集);
    撤销选择当前位置;
    dfs(下一位置, 路径(当前结果), 结果集);
}

每一次独立的决策都对应了多种选择（通常对应了每次决策能选择什么，或者每次决策能选择多少个 …）（对应到这类模板的题目有：17. 电话号码的数字组合、39. 组合总和 …）:
- 确定结束回溯过程的 base case
- 遍历所有的「选择」
- 对选择进行决策 (做选择 -> 递归 -> 撤销选择)

void dfs(选择列表, 路径(当前结果), 结果集) {
    if (满足结束条件) {
        结果集.add(路径);
        return;
    }
        
    for (选择 in 选择列表) {
        做选择;
        dfs(路径’, 选择列表, 结果集);
        撤销选择;
    }
}

类似题目

题目	思路
leetcode：77. 给定集合[1…n]，从中挑选k(指定)个数，返回所有组合(每个数可以用一次) combination	组合是顺序无关的，如 [1，2] 和 [2，1] 是同一个组合不同排列。组合时需要一个idx来排除已经选过的数：对于每个数，有两种选择，要，不要；当`path.size()==k`时时表示找到了一种组合
leetcode：216. 给定集合[1…9]，从中挑选k(指定)个数，令其和为target，返回所有组合(每个数可以用一次) combination-sum-iii	比77题多了一个限制，和为target。组合时需要一个idx来排除已经选过的数：对于每个数，有两种选择，要，不要；当path.size() == k && `currSum == targetSum`时表示找到了一种组合
leetcode：17. 给定一个数字到字母集的映射表，和一个数字组成的字符串，返回所有可能的组合 Letter Combinations of a Phone Number	组合是顺序无关的，如 [1，2] 和 [2，1] 是同一个组合不同排列。组合时需要一个idx来排除已经选过的数：从str[0]中选一个数(枚举所有可能的选择)，从str[1]中选择一个数… …当`idx==str.size()`时，表示找到了一种组合
leetcode：401. 二进制手表所有可能的表示时间 Binary Watch	时针集合取k个，分针集合取num-k个，然后将所有符合要求的生成时间表示存入结果中即可
leetcode：22. 给定n，生成所有合法的括号组合 generate-parentheses	因为要所有组合，所以应该回溯。从左到右尝试，对于当前位置，可以放`(`还是`)`
leetcode：39. 无序(不重复)数组中选出一些数，令其和=target，返回所有可能的组合(每个数可以使用无限次) Combination Sum	因为不知道要选几个数，所以不可以指针或者迭代；因为需要返回方案而不是方案个数，所以用回溯。对于每个数，有两种选择：不要、要(1、2…次)；当restT == 0时就说明找到了一条路径；当restT < 0或者`idx == N`时递归返回
leetcode：40. 无序(可重复)数组中选出一些数，令其和=target，返回所有可能的组合(每个数可以使用一次) Combination Sum II	和39题目一样，唯一的区别是需要去重。怎么去重了，在要了之后，要先跳过所有和nums[curr]相同的数字，再看要还是不要
leetcode：377. 无序(不重复)数组中选出一些数，令其和=target，返回所有可能的组合个数(每个数可以使用无限次) Combination Sum IV	对于每一个rest，在搜索开始之前，如果rest=0，说明找到了一种；否则每次均从`nums[0....]`开始使用每一个nums去拼接rest，如果rest > nums[idx]，那么就使用它；否则什么也不干
leetcode：254. 整数可以由因子相乘得来，给出一个整数，返回所有可能的因子组合	由于题目中说明了1和n本身不能算其因子，那么可以从2开始遍历到n，如果当前的数i可以被n整除，说明i是n的一个因子，将其存入一位数组 out 中，然后递归调用 n/i，此时不从2开始遍历，而是从i遍历到 n/i；停止的条件是当n等于1时，如果此时 out 中有因子，将这个组合存入结果 res 中（start排除已经选过的数，初始时start为2，最多为sqrt(n)）
leetcode：78. 无序(不重复)数组所有的不重复子集	对于每一个元素，都有选择和不选择两种选择，一直到没有元素可选了，才收集可能的答案
leetcode：90. 无序(可重复)数组所有的不重复子集 Subsets II	怎么去重呢？重复的原因是：刚刚选择了，然后撤销了这个选择，之后又选择了和刚刚相同的元素；所以先排序，然后去重（为什么要排序，将重复的元素放在一起，便于剪枝）
leetcode：46. 无序(不重复)数组所有的全排列 Permutations	数要全部用光(每个答案长度是固定的)，所以对于第一位可以选择num[0…x]，对于第二位可以选择除了第一位的所有选择…直到所有数全部用完
leetcode：47. 无序(可重复)数组所有的全排列(不重复) Permutations	怎么去重呢？对于两个相同的元素，可以两个都选，两个都不选，只选择一个（那么选哪一个都可以，因为和选择另一个是相同的情况，所以只有这种情况我们需要剪枝）。所以应该先排序，然后去重。去重条件是：和前一个元素值相同，并且前一个元素已经被使用过
leetcode：113. 二叉树找到路径(从根到叶)，令其和=target，返回所有可能的组合	先判断当前节点是否满足，然后去左右子树找，找完之后，返回上一个结点时，需要把该结点从path 中移除

OceanStar的学习笔记

关注

0
点赞
踩
0

收藏

觉得还不错? 一键收藏
0
评论
leetcode：216. 组合总和 III

题目解析回溯leetcode：77. 组合中要求：从集合[1…n]中选出k个数。本题比leetcode：77. 组合多了一个限制，和为n，即：从集合[1…9]中选出k个数，令其和为n。每个树只能用一次。本题k相当于了树的深度，9（因为整个集合就是9个数）就是树的宽度。例如 k = 2，n = 4的话，就是在集合[1,2,3,4,5,6,7,8,9]中求 k（个数） = 2, n（和） = 4的组合。选取过程如图：三部曲：（1）确定递归函数的参数：需要两个全局遍历vector&l
复制链接

扫一扫