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<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 中移除
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
给定一个整数数组 nums 和一个目标值 target,要求在数组中找出两个数的和等于目标值,并返回这两个数的索引。 思路1:暴力法 最简单的思路是使用两层循环遍历数组的所有组合,判断两个数的和是否等于目标值。如果等于目标值,则返回这两个数的索引。 此方法的时间复杂度为O(n^2),空间复杂度为O(1)。 思路2:哈希表 为了优化时间复杂度,可以使用哈希表来存储数组中的元素和对应的索引。遍历数组,对于每个元素nums[i],我们可以通过计算target - nums[i]的值,查找哈希表中是否存在这个差值。 如果存在,则说明找到了两个数的和等于目标值,返回它们的索引。如果不存在,将当前元素nums[i]和它的索引存入哈希表中。 此方法的时间复杂度为O(n),空间复杂度为O(n)。 思路3:双指针 如果数组已经排序,可以使用双指针的方法来求解。假设数组从小到大排序,定义左指针left指向数组的第一个元素,右指针right指向数组的最后一个元素。 如果当前两个指针指向的数的和等于目标值,则返回它们的索引。如果和小于目标值,则将左指针右移一位,使得和增大;如果和大于目标值,则将右指针左移一位,使得和减小。 继续移动指针,直到找到两个数的和等于目标值或者左指针超过了右指针。 此方法的时间复杂度为O(nlogn),空间复杂度为O(1)。 以上三种方法都可以解决问题,选择合适的方法取决于具体的应用场景和要求。如果数组规模较小并且不需要考虑额外的空间使用,则暴力法是最简单的方法。如果数组较大或者需要优化时间复杂度,则哈希表或双指针方法更合适。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值