代码随想录算法训练营第二十四天| 77. 组合
文章目录
- 1.回溯理论基础
- 2.组合
1.回溯理论基础
回溯法,一般可以解决如下几种问题:
- 组合问题:N个数里面按一定规则找出k个数的集合
- 切割问题:一个字符串按一定规则有几种切割方式
- 子集问题:一个N个数的集合里有多少符合条件的子集
- 排列问题:N个数按一定规则全排列,有几种排列方式
- 棋盘问题:N皇后,解数独等等
回溯法模板:
void backtracking(参数) {if (终止条件) {存放结果;return;}for (选择:本层集合中元素(树中节点孩子的数量就是集合的大小)) {处理节点;backtracking(路径,选择列表); // 递归回溯,撤销处理结果}
}
2.组合
给定两个整数 n 和 k,返回范围 [1, n] 中所有可能的 k 个数的组合。
回溯法解决的问题都可以抽象为树形结构(N叉树),你可以按 任何顺序 返回答案。
示例 1:
输入:n = 4, k = 2
输出:[ [2,4], [3,4], [2,3], [1,2], [1,3], [1,4], ]
示例 2:
输入:n = 1, k = 1
输出:[[1]]
提示:
1 <= n <= 201 <= k <= n
我们用回溯法解决,回溯法解决的问题都可以抽象为树形结构(N叉树)如下图:

本题我们先从
[1,n]中取一个数,再从剩下的集合中取一个数,一直操作,直到取得数字能凑够k个。此时得到一个集合,然后开始回溯一个数字,取这个数字下一个位置的数字,一直这样递归遍历下去。
回溯法三部曲:
递归函数的返回值以及参数:在这里要定义两个全局变量,一个用来存放符合条件单一结果path,一个用来存放符合条件结果的集合result
要有一个起点来作为下一个数字的标记,所以void backtracking(int n, int k, int startnode);
回溯函数终止条件:path满k即为终止,将path放到result中;
单层搜索的过程:for循环每次从startIndex开始遍历,然后用path保存取到的节点i。
代码如下:
class Solution {
private:vector<vector<int>> result;vector<int> path;void backtracking(int n, int k, int startnode) {if (path.size() == k) {//终止条件result.push_back(path);return;}for (int i = startnode; i <= n; i++) {//单层搜索逻辑path.push_back(i);backtracking(n, k, i + 1);path.pop_back();}}
public:vector<vector<int>> combine(int n, int k) {backtracking(n, k, 1);return result;}
};