77. 组合
问题描述:传送门
思路:
比较傻的想法是两个for循环,但是如果条件是n为100,k为50,
光for循环就需要50次,运算量超大。
所以回溯法引入进来,用递归来做层叠嵌套。
每一次的递归中嵌套一个for循环,那么递归就可以用于解决多层嵌套循环的问题了。
接下来用图例解释本题的回溯搜索过程:
第一次取1,集合变为2,3,4 ,因为k为2,我们只需要再取一个数就可以了,分别取2,3,4,得到集合 [1,2] [1,3] [1,4],以此类推。
从图中也可以发现,n相当于树的宽度,k相当于树的深度。
当把达到叶子节点的结果收集起来,就可以求得n个数中k个数的组合问题。
开始上菜:
1、确定回溯参数和返回值
参数:
一个存放符合条件的单一结果path ;
一个存放符合条件结果的集合result ;
n和k;
用于记录集合从哪里遍历的startIndex;
在本题里,比如选好1后,用startIndex来记录下一层递归,搜索的起始位置2,3,4。
vector<vector<int>> result; // 存放符合条件结果的集合
vector<int> path; // 用来存放符合条件单一结果
void backtracking(int n, int k, int startIndex)
2、回溯函数终止条件
path这个数组的大小如果为k,说明我们找到了一个子集大小为k的组合了.
如下图红色标记部分:
此时用result二维数组,把path保存起来,并终止本层递归。
所以终止条件代码如下:
if (path.size() == k) {
result.push_back(path);
return;
}
for循环每次从startIndex开始遍历,
然后用path来记录当前取得的startIndex节点。
代码如下:
for (int i = startIndex; i <= n; i++) { // 控制树的横向遍历
path.push_back(i); // 处理当前节点
backtracking(n, k, i + 1); // 递归:控制树的纵向遍历,下一层搜索要从i+1开始
path.pop_back();
// 回溯,撤销,for循环时,目前处理到的节点如【1,2】中的2,强制i++到3.
然后放入path。
}
可以看出backtracking(递归函数)通过不断调用自己一直往深处遍历,遇到了叶子节点就返回。
而backtracking()后的代码,就是回溯的操作了,即撤销本次处理的结果。
代码:
完整代码如下:
class Solution {
private:
vector<vector<int>> result; // 存放符合条件结果的集合
vector<int> path; // 用来存放符合条件结果
void backtracking(int n, int k, int startIndex) {
if (path.size() == k) {
result.push_back(path);
return;
}
for (int i = startIndex; 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;
}
};