题目来源
题目描述
题目解析
回溯
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 中移除 |