题目:216. 组合总和 III - 力扣(LeetCode)
思路:
1.递归函数参数和返回值:首先需要两个全局变量一维path数组和二维result数组,path数组用来收集路径上的元素,result数组用来存放符合条件的结果集 。
参数1:targetSum即题目要求的元素和;参数2:k即满足元素和的集合个数;参数3:sum用来统计已经收集的元素的和,即path数组里的元素和;参数4:startIndex为下一层for循环搜索找到起始位置。
2.终止条件:如果path数组的元素个数等于k,并且sum == targetSum,说明找到了一个path集合放进result结果集里;如果path数组的元素个数等于k,并且sum != targetSum,直接返回空。
3.单层搜索过程:path用来收集路径上的元素,sum用来统计路径上的元素和,然后递归进入下一次循环,同时回溯sum和path数组。
代码:
class Solution {
public:
vector<int> path;
vector<vector<int>> result;
void backtracking(int sum, int k, int targetSum, int startIndex){
if(path.size() == k){
if(targetSum == sum) result.push_back(path);
return;// 如果path.size() == k 但sum != targetSum 直接返回
}
for(int i = startIndex; i <= 9; i++){
path.push_back(i);//收集路径上的元素
sum += i;//统计路径上的元素和
backtracking(sum, k, targetSum, i + 1);//递归,进入下一层循环
sum -= i;//回溯路径上的元素和
path.pop_back();//回溯收集路径上的元素
}
}
vector<vector<int>> combinationSum3(int k, int n) {
path.clear();
result.clear();
backtracking(0, k, n, 1);
return result;
}
};
剪枝操作:
如果sum > targetSum,直接返回就行了,不用继续往下遍历到path数组元素等于k个,再判断元素和不等于targetSum。
代码:
class Solution {
public:
vector<int> path;
vector<vector<int>> result;
void backtracking(int sum, int k, int targetSum, int startIndex){
if(sum > targetSum){
return;//如果sum > targetSum,直接返回就行了,不用继续往下遍历到path数组元素等于k个,再判断元素和不等于targetSum
}
if(path.size() == k){
if(targetSum == sum) result.push_back(path);
return;// 如果path.size() == k 但sum != targetSum 直接返回
}
for(int i = startIndex; i <= 9; i++){
path.push_back(i);//收集路径上的元素
sum += i;//统计路径上的元素和
backtracking(sum, k, targetSum, i + 1);//递归,进入下一层循环
sum -= i;//回溯路径上的元素和
path.pop_back();//回溯收集路径上的元素
}
}
vector<vector<int>> combinationSum3(int k, int n) {
path.clear();
result.clear();
backtracking(0, k, n, 1);
return result;
}
};
题目: 17. 电话号码的字母组合 - 力扣(LeetCode)
思路:
1.递归参数和返回值:首先需要一个字符串s来收集叶子节点的结果,然后用一个字符串数组result保存起来,参数一个是给定的string digits,然后还要有一个参数就是int型的index,这个index是记录遍历第几个数字了,就是用来遍历digits的(题目中给出数字字符串),同时index也表示树的深度。
2.终止条件:就是index 等于输入的数字个数(digits.size)。index从0开始计数,如果digits中元素个数为2,当index = 0时,第一层递归:收集digits[0]中的元素,index = 1时,第二层递归:收集digits[1]中的元素,再往下递归,此时index = 2等于digits中元素个数,此时已经把digits遍历完了,不需要收集了,所以直接返回。
3.单层搜索过程:首先要取index指向的数字,并找到对应的字符集(手机键盘的字符集)。然后收集字符集中的元素,递归再回溯。
代码:
class Solution {
public:
const string letters[10] = {//对数字到字母做一个字符串映射
"",//0
"",//1
"abc",//2
"def",//3
"ghi",//4
"jkl",//5
"mno",//6
"pqrs",//7
"tuv",//8
"wxyz"//9
};
string path;//字符串path用来收集路径上的字符元素
vector<string> result;//result数组存放结果集
void backtracking(const string& digits, int index){//参数index是记录遍历第几个数字了,就是用来遍历digits的
if(digits.size() == index){//index从0开始计数,如果digits中元素个数为2,当index = 0时,第一层递归:收集digits[0]中的元素,index = 1时,第二层递归:收集digits[1]中的元素,再往下递归,此时index = 2等于digits中元素个数,此时已经把digits遍历完了,不需要收集了,所以直接返回.
result.push_back(path);
return;
}
int digit = digits[index] - '0';//将index指向的数字转为int
string letter = letters[digit];//取数字对应的字符串
for(int i = 0; i < letter.size(); i++){
path.push_back(letter[i]);//搜集元素
backtracking(digits, index + 1);//递归,注意index+1,下一层要处理digits的下一个数字了
path.pop_back();//回溯
}
return;
}
vector<string> letterCombinations(string digits) {
path.clear();
result.clear();
if(digits.size() == 0) return result;
backtracking(digits, 0);
return result;
}
};