力扣 216 组合总和3 【回溯三步法】

65 篇文章 1 订阅

力扣 216 组合总和3

全部刷题与学习记录

【C++刷题学习笔记目录】

【C++百万并发网络通信-笔记目录】

原题目

题目地址:216. 组合总和 III

找出所有相加之和为 n的 k 个数的组合。组合中只允许含有 1 - 9 的正整数,并且每种组合中不存在重复的数字。

要求:1、所有数字都是正整数 2、解集不能包含重复组合

示例 1:

输入: k = 3, n = 7
输出: [[1,2,4]]

示例 2:

输入: k = 3, n = 9
输出: [[1,2,6], [1,3,5], [2,3,4]]

考查知识点

回溯三步法、回溯开始的地方(startIndex的每轮取值)


自己的第一遍解法

其实做完了回溯法的入门题目力扣 77 组合【回溯经典问题】【三步法】以后,再看这道题就不会发怵了,但是要在细节上把握住。

首先就是回溯三步法:

1、回溯函数参数及返回值:很明显除了题目的kn以外,需要一个startIndex,因为这同样是一个组合问题

2、回溯函数终止条件:1)满足题目要求的pathpath是满足题目要求的一个结果,即多叉树中的一条路径)一定有pathSum == n && path.size() == k,其中pathSumpath中所有数字的总和;2)不满足题目要求还要终止的有pathSum != n && path.size() == k

3、单层回溯函数逻辑:首先要将当前数字加入路径path,累加pathSum,接下来就是递归backTracking(),最后一定注意撤销,这里撤销除了path要弹出尾元素之外,还要撤销pathSum的上一次累加

class Solution {
private:
    vector<vector<int>> result;
    vector<int> path;
    int pathSum = 0;
    void backTracking(int k, int n, int startIndex) { //1:回溯函数参数及返回值
        if (pathSum == n && path.size() == k) { //2:回溯函数终止条件
            result.push_back(path);
            return;
        }
        else if (path.size() == k)
            return;
        if (pathSum > n) return;//剪枝

        for (int i = startIndex; i <= 9; ++i) { //3:单层逻辑
            path.push_back(i);
            pathSum += path.back();
            backTracking(k, n, i + 1); //注意下次startIndex的选取
            pathSum -= path.back(); //:回溯撤销
            path.pop_back();
        }

    }
public:
    vector<vector<int>> combinationSum3(int k, int n) {
        backTracking(k, n, 1);

        return result;
    }

};

程序中还进行了剪枝,剪枝不光可以在for循环变量这里做文章,还可以在满足一定条件时直接return,本文就是这样。如果不进行剪枝的话,力扣就会超时。

最后,回溯三步法汇总还需要注意的一个细节就是for循环中调用回溯函数backTracking()时传入的参数:startIndex,一定是 i+1,表示下一次选取数字的一定>当前数字,这里很重要,这也是组合问题的体现。

还有本题限制了每个数字必须在[1, 9]中,因此for循环变量i的最大取值就是9,不能写成n

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
力扣是一个在线编程平台,提供了大量的算法题目,可以帮助程序员提高算法能力。回溯算法是一种搜索算法,它通过不断地尝试所有可能的解来求解问题。在回溯算法中,我们首先定义一个解空间,然后从解空间中搜索所有可能的解,直到找到符合要求的解为止。回溯算法通常用于求解组合问题、排列问题、子集问题等。 在 Java 中实现回溯算法,通常需要定义一个递归函数来搜索解空间。在递归函数中,我们首先判断当前状态是否符合要求,如果符合要求,则将当前状态加入到解集中;否则,我们继续搜索下一个状态。在搜索下一个状态时,我们需要对当前状态进行一些修改,然后递归调用自身来搜索下一个状态。当搜索完所有可能的状态后,我们需要回溯到上一个状态,继续搜索其他可能的状态。 以下是回溯算法的一般步骤: 1. 定义解空间:确定问题的解空间,并定义一个数据结构来表示解空间中的每个状态。 2. 确定约束条件:确定哪些状态是合的,并定义一个函数来判断当前状态是否符合要求。 3. 确定搜索策略:确定搜索解空间的顺序,并定义一个函数来生成下一个状态。 4. 搜索解空间:使用递归函数搜索解空间,如果当前状态符合要求,则将其加入到解集中;否则,继续搜索下一个状态。 5. 回溯:当搜索完所有可能的状态后,回溯到上一个状态,继续搜索其他可能的状态。 以下是一个力扣题目的回溯算法 Java 实现示例: ``` class Solution { List<List<Integer>> res = new ArrayList<>(); List<Integer> path = new ArrayList<>(); public List<List<Integer>> subsets(int[] nums) { dfs(nums, 0); return res; } private void dfs(int[] nums, int start) { res.add(new ArrayList<>(path)); for (int i = start; i < nums.length; i++) { path.add(nums[i]); dfs(nums, i + 1); path.remove(path.size() - 1); } } } ``` 该算法用于求解给定数组的所有子集。在递归函数中,我们首先将当前状态加入到解集中,然后从当前位置开始搜索下一个状态。在搜索下一个状态时,我们将当前元素加入到路径中,并递归调用自身来搜索下一个状态。当搜索完所有可能的状态后,我们需要回溯到上一个状态,继续搜索其他可能的状态。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值