LeetCode 216- 组合总和III
题目链接:力扣
题目描述:找出所有相加之和为 n 的 k 个数的组合,且满足下列条件:
只使用数字1到9
每个数字 最多使用一次
返回 所有可能的有效组合的列表 。该列表不能包含相同的组合两次,组合可以以任何顺序返回
解题思路
- 经典的回溯题目,本题要求我们不能重复取集合。所以我们得额外使用一个startIndex来控制每次取数的开始位置,防止出现重复,然后又要求取k个数,所以我们应该在终止条件处加判断当取够数字时收取结果。
class Solution {
public:
vector<int> path;
vector<vector<int>> result;
void backtracking(int k,int startIndex,int target){
//target<0证明数字太大了,直接剪枝
if(target < 0) return;
//当取够数字时,直接返回,但是否收取结果还要看target的值
if(path.size() == k){
if(0 == target) result.push_back(path);
return ;
}
//此处继续剪枝
for(int i=startIndex;i<=9-(k-path.size())+1;i++){
path.push_back(i);
backtracking(k,i+1,target - i);//隐藏target+i的回溯
path.pop_back();
}
}
vector<vector<int>> combinationSum3(int k, int target) {
backtracking(k,1,target);
return result;
}
};
总结:
- 这题剪枝操作可以这样理解
-
- 当前取 path.size()
- 还要取k-path.size()
- 我们取数的区间是[i,n],一共有 n-i+1 个数,那么剩下的数一定要大于等于还要取的数即
n-i+1 >= k-path.size() =====》 i<=n-(k-path.size())+1
所以有了如上的剪枝操作,我们每次进行剪枝一般都是在for循环中下功夫
LeetCode 17- 电话号码的字母组合
题目链接:力扣
题目描述:给定一个仅包含数字 2-9 的字符串,返回所有它能表示的字母组合。答案可以按 任意顺序 返回。
给出数字到字母的映射如下(与电话按键相同)。注意 1 不对应任何字母。
解题思路
- 首先将数字和对应的字母映射到一个二维数组中去,再通过回溯递归来收集结果。
- 依靠for循环来选取每个数字对应字母,index来控制递归深度。
class Solution {
public:
vector<string> result;
string s;
const string letterMap[10] = {
"", // 0
"", // 1
"abc", // 2
"def", // 3
"ghi", // 4
"jkl", // 5
"mno", // 6
"pqrs", // 7
"tuv", // 8
"wxyz", // 9
};
//index用来控制遍历到digits中哪个数字了,
void backtracking(const string &digits,int index){
//因为如果正好是最后一个数字,那么下面还要进行处理。
//当是最后一个数字的下一个数字的时候才会收集结果。
if(index == digits.size()){
result.push_back(s);
return ;
}
//将字符转换为对应的数字
int digit = digits[index] - '0';
//取出对应数字的字母。
string letter = letterMap[digit];
//对数字中每个字母进行遍历,当第一次进去时,也就是取第一个数字的第一个字母,
//然后再循环中又进一层递归,就是取第二个数字的第一个字母,然后出,也就是弹出第二个数字
//的第一个字母,再取第二个数字的第二个字母递归回溯来完成。
for(int i = 0;i<letter.size();i++){
s.push_back(letter[i]);
backtracking(digits,index+1);
s.pop_back();
}
}
vector<string> letterCombinations(string digits) {
if(digits.size() == 0) return result;
backtracking(digits,0);
return result;
}
};
总结:
- 看到这题大概知道怎么办,但是就是不知道怎么控制每个数字中字符遍历的位置,结果不用控制,直接for循环遍历就行。
- 并且由于我们不是在一个集合中反复取数,所以不需要进行去重操作,也就是不需要startIndex。前几题都是在一个集合中取数字,所以需要startIndex来控制,防止重复了。